test without pandas

nogibjj · Nov 22, 2024 · 154a052 · 154a052
1 parent 9d4cde6
commit 154a052
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/test_main.py b/test_main.py
@@ -17,16 +17,18 @@
 def test_extract_load():
 
     output_path = extract_load(url="https://shorturl.at/5YexG", 
-            file_path="data/heart-failure.csv")
+    file_path="data/heart-failure.csv",
+    directory="data")
     assert os.path.exists(output_path)
 
 def test_transform():
-    df = transform_table("heart-failure-load","heart-failure-transform")
+    df = transform_table("heart_failure_load","heart-failure-transform")
     assert df is None, "Dataframe is None; load_data failed."
 
 def test_query():
     extract_load(url="https://shorturl.at/5YexG", 
-            file_path="data/heart-failure.csv")
+    file_path="data/heart-failure.csv",
+    directory="data")
     transform_table("heart-failure-load","heart-failure-transform")
     assert query("SELECT AVG(age) FROM heart-failure-transform WHERE smoking=1") is None