Use built-in unittest.mock (#197)

grove/tests/amplification/test_grover.py

@@ -1,6 +1,6 @@
 import numpy as np
 import pytest
-from mock import patch
+from unittest.mock import patch
 from pyquil import Program
 from pyquil.gates import H, X
 from pyquil.quilatom import QubitPlaceholder

grove/tests/bernstein_vazirani/test_bernstein_vazirani.py

@@ -1,5 +1,5 @@
 import numpy as np
-from mock import patch
+from unittest.mock import patch
 
 from grove.bernstein_vazirani.bernstein_vazirani import BernsteinVazirani, create_bv_bitmap
 

grove/tests/circuit_primitives/test_swap.py

@@ -3,7 +3,7 @@
 """
 import numpy as np
 import pytest
-from mock import patch
+from unittest.mock import patch
 from pyquil import Program
 from pyquil.gates import CSWAP, H
 

grove/tests/deutsch_jozsa/test_deutsch_jozsa.py

@@ -1,6 +1,6 @@
 import numpy as np
 import pytest
-from mock import patch
+from unittest.mock import patch
 from pyquil import Program
 from pyquil.gates import X, H, CNOT
 

grove/tests/ising/test_ising.py

@@ -1,5 +1,5 @@
 import numpy as np
-from mock import patch, Mock
+from unittest.mock import patch, Mock
 from pyquil.api import WavefunctionSimulator
 
 from grove.ising.ising_qaoa import energy_value, ising

grove/tests/jordan_gradient/test_jordan_gradient.py

@@ -1,5 +1,5 @@
 import numpy as np
-from mock import patch
+from unittest.mock import patch
 from pyquil import Program
 from pyquil.gates import H, CPHASE, SWAP, MEASURE
 

grove/tests/measurements/test_estimation.py

@@ -3,7 +3,7 @@
 """
 import numpy as np
 import pytest
-from mock import Mock, patch
+from unittest.mock import Mock, patch
 from pyquil import Program
 from pyquil.api import QuantumComputer
 from pyquil.gates import RY, RX, I

grove/tests/pyqaoa/test_maxcut.py

@@ -16,7 +16,7 @@
 import numpy as np
 import pytest
 import pyquil.api as qc_mod
-from mock import Mock, patch
+from unittest.mock import Mock, patch
 from pyquil import Program
 from pyquil.gates import H, X, PHASE, CNOT, RZ
 from pyquil.paulis import PauliTerm, PauliSum

grove/tests/pyqaoa/test_qaoa.py

@@ -15,7 +15,7 @@
 ##############################################################################
 
 import numpy as np
-from mock import Mock, patch
+from unittest.mock import Mock, patch
 from pyquil import Program
 from pyquil.api import QuantumComputer, WavefunctionSimulator
 from pyquil.gates import X, Y, Z

grove/tests/pyvqe/test_algorithms.py

@@ -18,7 +18,7 @@
 import funcsigs
 import numpy as np
 import pytest
-from mock import Mock, MagicMock, patch
+from unittest.mock import Mock, MagicMock, patch
 from pyquil import Program
 from pyquil.api import WavefunctionSimulator
 from pyquil.gates import H, RX, RZ

grove/tests/simon/test_simon.py

@@ -3,7 +3,7 @@
 from os.path import abspath, dirname
 
 import numpy as np
-from mock import patch
+from unittest.mock import patch
 from pyquil import Program
 
 from grove.simon.simon import Simon, create_1to1_bitmap, create_valid_2to1_bitmap

grove/tests/tomography/test_process_tomography.py

@@ -17,8 +17,7 @@
 import pytest
 import os
 import numpy as np
-from mock import patch
-from mock import MagicMock, Mock
+from unittest.mock import patch, MagicMock, Mock
 import json
 
 from pyquil.api import Job, QVMConnection

grove/tests/tomography/test_state_tomography.py

@@ -20,8 +20,7 @@
 import numpy as np
 import pytest
 
-from mock import MagicMock
-from mock import patch
+from unittest.mock import MagicMock, patch
 from pyquil.api import Job, QVMConnection
 
 from grove.tomography.operator_utils import make_diagonal_povm

grove/tests/tomography/test_utils.py

@@ -17,7 +17,7 @@
 import numpy as np
 import pytest
 from matplotlib.pyplot import figure
-from mock import Mock, patch, call
+from unittest.mock import Mock, patch, call
 from mpl_toolkits.mplot3d import Axes3D
 from pyquil.api import QuantumComputer
 from pyquil.gates import X, Y, I

requirements.txt

@@ -14,7 +14,6 @@ jupyter_contrib_nbextensions
 # test dependencies
 tox
 pytest >= 3.0.0
-mock
 pytest-cov
 codeclimate-test-reporter
 

setup.py

@@ -50,7 +50,6 @@
     tests_require=[
         'tox',
         'pytest >= 3.0.0',
-        'mock'
     ],
     license='LICENSE',
     keywords='quantum quil programming hybrid'