move deprecated_test to _compat_test (#378)

openfermioncirq/_compat.py

@@ -9,33 +9,10 @@
 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
-from typing import Any, Callable, Dict, Tuple
+from typing import Dict, Tuple
 from types import ModuleType
-import functools
 import warnings
 
-import pytest
-
-
-def deprecated_test(test: Callable) -> Callable:
-    """Marks a test as using deprecated functionality.
-
-    Ensures the test is executed within the `pytest.deprecated_call()` context.
-
-    Args:
-        test: The test.
-
-    Returns:
-        The decorated test.
-    """
-
-    @functools.wraps(test)
-    def decorated_test(*args, **kwargs) -> Any:
-        with pytest.deprecated_call():
-            test(*args, **kwargs)
-
-    return decorated_test
-
 
 def wrap_module(module: ModuleType,
                 deprecated_attributes: Dict[str, Tuple[str, str]]):

openfermioncirq/_compat_test.py

@@ -9,13 +9,35 @@
 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
+from typing import Any, Callable
+import functools
 import warnings
 
 import pytest
 import deprecation
 
 import openfermioncirq as ofc
-from openfermioncirq._compat import deprecated_test, wrap_module
+from openfermioncirq._compat import wrap_module
+
+
+def deprecated_test(test: Callable) -> Callable:
+    """Marks a test as using deprecated functionality.
+
+    Ensures the test is executed within the `pytest.deprecated_call()` context.
+
+    Args:
+        test: The test.
+
+    Returns:
+        The decorated test.
+    """
+
+    @functools.wraps(test)
+    def decorated_test(*args, **kwargs) -> Any:
+        with pytest.deprecated_call():
+            test(*args, **kwargs)
+
+    return decorated_test
 
 
 @deprecation.deprecated()

openfermioncirq/gates/common_gates_test.py

@@ -18,7 +18,7 @@
 
 import cirq
 import openfermioncirq as ofc
-from openfermioncirq._compat import deprecated_test
+from openfermioncirq._compat_test import deprecated_test
 
 
 def test_fswap_interchangeable():

openfermioncirq/gates/three_qubit_gates_test.py

@@ -19,7 +19,7 @@
 from cirq.testing import EqualsTester
 
 import openfermioncirq as ofc
-from openfermioncirq._compat import deprecated_test
+from openfermioncirq._compat_test import deprecated_test
 
 
 @deprecated_test