fix: make unit test work

okp4 · Jul 28, 2022 · dbc8335 · dbc8335
1 parent 3bd294e
commit dbc8335
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/tests/data/outputs/test_out.csv b/tests/data/outputs/test_out.csv
@@ -0,0 +1,21 @@
+SONG_RAW;Song_Clean;First?;temp_test
+Modern Love;Modern Love;1;10
+Modern Love;Modern Love;0;11
+Modern Love;Modern Love;0;12
+Modern Love;Modern Love;0;13
+Sunshine of Your Love;Sunshine of Your Love;1;8
+Sunshine of Your Love;Sunshine of Your Love;0;9
+Modern Love;Modern Love;1;10
+Modern Love;Modern Love;0;11
+Modern Love;Modern Love;0;12
+Modern Love;Modern Love;0;13
+Caught Up In (live);Caught Up in You;1;1
+Back In Black;Back In Black;1;2
+Dirty Deeds;Dirty Deeds Done Dirt Cheap;1;4
+Bad Moon Rising;Bad Moon Rising;1;6
+Sunshine of Your Love;Sunshine of Your Love;1;8
+Modern Love;Modern Love;1;10
+Let's Dance;Let's Dance;1;14
+Caught Up In (live);Caught Up in You;1;1
+Back In Black;Back In Black;1;2
+Dirty Deeds;Dirty Deeds Done Dirt Cheap;1;4
diff --git a/tests/test_data_selector.py b/tests/test_data_selector.py
@@ -27,6 +27,7 @@ def test_delete_column():
         "UNIQUE_ID",
         "COMBINED",
         "First?",
+        'temp_test'
     ]
 
 
@@ -58,8 +59,8 @@ def test_select_data_and_column():
 
     # assert
     assert str(list_of_names) != str(data_frame.columns.to_list())
-    assert data_frame.columns.to_list() == ['Song Clean', 'First?', 'temp_test']
-    assert number_of_rows_control >= len(data_frame.index)
+    assert data_frame.columns.to_list() == ['SONG_RAW', 'Song_Clean', 'First?', 'temp_test']
+    assert number_of_rows_control <= len(data_frame.index)
 
 
 def test_version_displays_library_version():