fix marco-c#68

anujanegi · Mar 28, 2018 · 810a89c · 810a89c
1 parent 324866f
commit 810a89c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -52,6 +52,7 @@ def test_write_labels(tmpdir):
     file_path = tmpdir.join('test.csv')
     utils.write_labels(label, file_name=file_path)
     assert(os.path.exists(file_path))
+    assert(label == utils.read_labels(file_name=file_path))
 
 
 test_balance_data = [