0.6.15

mabel-dev · Oct 15, 2023 · 914966b · 914966b
1 parent be4e5c4
commit 914966b
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 11 deletions.
diff --git a/tests/storage_tests/google/test_reader_gcs_reader.py b/tests/storage_tests/google/test_reader_gcs_reader.py
@@ -19,6 +19,6 @@ def test_blockers():
 
 
 if __name__ == "__main__":  # pragma: no cover
-    test_blockers()
+    from tests.helpers.runner import run_tests
 
-    print("okay")
+    run_tests()
diff --git a/tests/storage_tests/google/test_rw_gcs.py b/tests/storage_tests/google/test_rw_gcs.py
@@ -42,6 +42,7 @@ def test_gcs_binary():
         project="testing",
         blob_size=1024,
         dataset=f"{BUCKET_NAME}/test/gcs/dataset/binary",
+        schema=["index"],
     )
     for i in range(200):
         w.append({"index": i + 300})
@@ -69,6 +70,7 @@ def test_gcs_text():
         blob_size=1024,
         format="jsonl",
         dataset=f"{BUCKET_NAME}/test/gcs/dataset/text",
+        schema=["index"],
     )
     for i in range(250):
         w.append({"index": i + 300})
@@ -87,7 +89,6 @@ def test_gcs_text():
 
 
 if __name__ == "__main__":  # pragma: no cover
-    test_gcs_binary()
-    test_gcs_text()
+    from tests.helpers.runner import run_tests
 
-    print("okay")
+    run_tests()
diff --git a/tests/storage_tests/google/test_rw_gcs_parquet.py b/tests/storage_tests/google/test_rw_gcs_parquet.py
@@ -35,6 +35,7 @@ def test_gcs_parquet():
             inner_writer=GoogleCloudStorageWriter,
             format="parquet",
             dataset=f"{BUCKET_NAME}/test/gcs/dataset",
+            schema=["$$"],
         )
         for i in range(100):
             w.append({"$$": i * 300})
@@ -54,6 +55,6 @@ def test_gcs_parquet():
 
 
 if __name__ == "__main__":  # pragma: no cover
-    test_gcs_parquet()
+    from tests.helpers.runner import run_tests
 
-    print("okay")
+    run_tests()
diff --git a/tests/storage_tests/minio_s3/test_reading_public_s3.py b/tests/storage_tests/minio_s3/test_reading_public_s3.py
@@ -30,6 +30,6 @@ def test_reading_aws_using_minio():
 
 
 if __name__ == "__main__":  # pragma: no cover
-    test_reading_aws_using_minio()
+    from tests.helpers.runner import run_tests
 
-    print("okay")
+    run_tests()
diff --git a/tests/storage_tests/minio_s3/test_rw_minio.py b/tests/storage_tests/minio_s3/test_rw_minio.py
@@ -48,6 +48,7 @@ def test_using_batch_writer():
             secret_key=os.getenv("MINIO_SECRET_KEY"),
             secure=False,
             dataset=f"{BUCKET_NAME}/test_writer",
+            schema=["player", "from"],
         )
 
         for member in VAMPIRIC_COUNCIL:
@@ -61,6 +62,6 @@ def test_using_batch_writer():
 
 
 if __name__ == "__main__":  # pragma: no cover
-    test_using_batch_writer()
+    from tests.helpers.runner import run_tests
 
-    print("okay")
+    run_tests()