Migrate test_external_class

Author: ethho

tests/test_external_class.py

@@ -1,54 +1,47 @@
-from nose.tools import assert_true, assert_list_equal
 from numpy.testing import assert_almost_equal
 import datajoint as dj
-from . import schema_external as modu
+from . import schema_external
 
 
-def setUp(self):
-    dj.config["stores"] = modu.stores_config
+def test_heading(schema_ext, mock_stores):
+    heading = schema_external.Simple().heading
+    assert "item" in heading
+    assert heading["item"].is_external
 
 
-def test_heading():
-    heading = modu.Simple().heading
-    assert_true("item" in heading)
-    assert_true(heading["item"].is_external)
-
-
-def test_insert_and_fetch():
+def test_insert_and_fetch(schema_ext, mock_stores, mock_cache):
     original_list = [1, 3, 8]
-    modu.Simple().insert1(dict(simple=1, item=original_list))
+    schema_external.Simple().insert1(dict(simple=1, item=original_list))
     # test fetch
-    q = (modu.Simple() & {"simple": 1}).fetch("item")[0]
-    assert_list_equal(list(q), original_list)
+    q = (schema_external.Simple() & {"simple": 1}).fetch("item")[0]
+    assert list(q) == original_list
     # test fetch1 as a tuple
-    q = (modu.Simple() & {"simple": 1}).fetch1("item")
-    assert_list_equal(list(q), original_list)
+    q = (schema_external.Simple() & {"simple": 1}).fetch1("item")
+    assert list(q) == original_list
     # test fetch1 as a dict
-    q = (modu.Simple() & {"simple": 1}).fetch1()
-    assert_list_equal(list(q["item"]), original_list)
+    q = (schema_external.Simple() & {"simple": 1}).fetch1()
+    assert list(q["item"]) == original_list
     # test without cache
     previous_cache = dj.config["cache"]
     dj.config["cache"] = None
-    q = (modu.Simple() & {"simple": 1}).fetch1()
-    assert_list_equal(list(q["item"]), original_list)
+    q = (schema_external.Simple() & {"simple": 1}).fetch1()
+    assert list(q["item"]) == original_list
     # test with cache
     dj.config["cache"] = previous_cache
-    q = (modu.Simple() & {"simple": 1}).fetch1()
-    assert_list_equal(list(q["item"]), original_list)
+    q = (schema_external.Simple() & {"simple": 1}).fetch1()
+    assert list(q["item"]) == original_list
 
 
-def test_populate():
-    image = modu.Image()
+def test_populate(schema_ext, mock_stores):
+    image = schema_external.Image()
     image.populate()
     remaining, total = image.progress()
-    assert_true(total == len(modu.Dimension() * modu.Seed()) and remaining == 0)
+    assert total == len(schema_external.Dimension() * schema_external.Seed()) and remaining == 0
     for img, neg, dimensions in zip(
-        *(image * modu.Dimension()).fetch("img", "neg", "dimensions")
+        *(image * schema_external.Dimension()).fetch("img", "neg", "dimensions")
     ):
-        assert_list_equal(list(img.shape), list(dimensions))
+        assert list(img.shape) == list(dimensions)
         assert_almost_equal(img, -neg)
     image.delete()
-    dj.errors._switch_filepath_types(True)
     for external_table in image.external.values():
         external_table.delete(display_progress=False, delete_external_files=True)
-    dj.errors._switch_filepath_types(False)