Update test_api.py

MacHu-GWU · May 28, 2024 · 2489d73 · 2489d73
1 parent a4505b7
commit 2489d73
Showing 1 changed file with 24 additions and 7 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -40,13 +40,30 @@ def test():
     _ = sam.patterns.status_tracker.Updates
 
     _ = sam.patterns.large_binary_column.aws_s3
-    from .aws_s3 import PutS3BackedColumnResult
-    from .aws_s3 import put_s3backed_column
-    from .aws_s3 import clean_up_created_s3_object_when_create_or_update_row_failed
-    from .aws_s3 import clean_up_old_s3_object_when_update_row_succeeded
-    from .aws_s3 import PutS3ApiCall
-    from .aws_s3 import PutS3Result
-    from .aws_s3 import put_s3
+    _ = sam.patterns.large_binary_column.aws_s3.PutS3BackedColumnResult
+    _ = sam.patterns.large_binary_column.aws_s3.put_s3backed_column
+    _ = (
+        sam.patterns.large_binary_column.aws_s3.clean_up_created_s3_object_when_create_or_update_row_failed
+    )
+    _ = (
+        sam.patterns.large_binary_column.aws_s3.clean_up_old_s3_object_when_update_row_succeeded
+    )
+    _ = sam.patterns.large_binary_column.aws_s3.PutS3ApiCall
+    _ = sam.patterns.large_binary_column.aws_s3.PutS3Result
+    _ = sam.patterns.large_binary_column.aws_s3.put_s3
+
+    _ = sam.patterns.large_binary_column.local
+    _ = sam.patterns.large_binary_column.local.WriteFileBackedColumnResult
+    _ = sam.patterns.large_binary_column.local.write_file_backed_column
+    _ = (
+        sam.patterns.large_binary_column.local.clean_up_new_file_when_create_or_update_row_failed
+    )
+    _ = (
+        sam.patterns.large_binary_column.local.clean_up_old_file_when_update_row_succeeded
+    )
+    _ = sam.patterns.large_binary_column.local.WriteFileApiCall
+    _ = sam.patterns.large_binary_column.local.WriteFileResult
+    _ = sam.patterns.large_binary_column.local.write_file
 
 
 if __name__ == "__main__":