Format with black

Author: ethho

tests/conftest.py

@@ -179,12 +179,19 @@ def stores_config(tmpdir_factory):
     stores_config = {
         "raw": dict(protocol="file", location=tmpdir_factory.mktemp("raw")),
         "repo": dict(
-            stage=tmpdir_factory.mktemp("repo"), protocol="file", location=tmpdir_factory.mktemp("repo")
+            stage=tmpdir_factory.mktemp("repo"),
+            protocol="file",
+            location=tmpdir_factory.mktemp("repo"),
         ),
         "repo-s3": dict(
-            S3_CONN_INFO, protocol="s3", location="dj/repo", stage=tmpdir_factory.mktemp("repo-s3")
+            S3_CONN_INFO,
+            protocol="s3",
+            location="dj/repo",
+            stage=tmpdir_factory.mktemp("repo-s3"),
+        ),
+        "local": dict(
+            protocol="file", location=tmpdir_factory.mktemp("local"), subfolding=(1, 1)
         ),
-        "local": dict(protocol="file", location=tmpdir_factory.mktemp("local"), subfolding=(1, 1)),
         "share": dict(
             S3_CONN_INFO, protocol="s3", location="dj/store/repo", subfolding=(2, 4)
         ),

tests/test_external.py

@@ -11,7 +11,9 @@ def test_external_put(schema_ext, mock_stores, mock_cache):
     """
     external storage put and get and remove
     """
-    ext = ExternalTable(schema_ext.connection, store="raw", database=schema_ext.database)
+    ext = ExternalTable(
+        schema_ext.connection, store="raw", database=schema_ext.database
+    )
     initial_length = len(ext)
     input_ = np.random.randn(3, 7, 8)
     count = 7
@@ -30,14 +32,15 @@ def test_external_put(schema_ext, mock_stores, mock_cache):
 
 
 class TestLeadingSlash:
-
     def test_s3_leading_slash(self, schema_ext, mock_stores, mock_cache, minio_client):
         """
         s3 external storage configured with leading slash
         """
         self._leading_slash(schema_ext, index=100, store="share")
 
-    def test_file_leading_slash(self, schema_ext, mock_stores, mock_cache, minio_client):
+    def test_file_leading_slash(
+        self, schema_ext, mock_stores, mock_cache, minio_client
+    ):
         """
         File external storage configured with leading slash
         """
@@ -50,42 +53,58 @@ def _leading_slash(self, schema_ext, index, store):
         id = index
         dj.config["stores"][store]["location"] = "leading/slash/test"
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         id = index + 1
         dj.config["stores"][store]["location"] = "/leading/slash/test"
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         id = index + 2
         dj.config["stores"][store]["location"] = "leading\\slash\\test"
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         id = index + 3
         dj.config["stores"][store]["location"] = "f:\\leading\\slash\\test"
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         id = index + 4
         dj.config["stores"][store]["location"] = "f:\\leading/slash\\test"
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         id = index + 5
         dj.config["stores"][store]["location"] = "/"
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         id = index + 6
         dj.config["stores"][store]["location"] = "C:\\"
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         id = index + 7
         dj.config["stores"][store]["location"] = ""
         SimpleRemote.insert([{"simple": id, "item": value}])
-        assert np.array_equal(value, (SimpleRemote & "simple={}".format(id)).fetch1("item"))
+        assert np.array_equal(
+            value, (SimpleRemote & "simple={}".format(id)).fetch1("item")
+        )
 
         dj.config["stores"][store]["location"] = oldConfig