New utils tests

SmokinCaterpillar · Feb 27, 2018 · a5ec11e · a5ec11e
1 parent 41ebf1f
commit a5ec11e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/integration_tests/utils_test.py b/integration_tests/utils_test.py
@@ -1,4 +1,7 @@
 import os
+import logging
+
+import pandas as pd
 
 from integration_tests.bchain.getdata_test import temp_dir
 
@@ -21,3 +24,10 @@ def test_clean_up(temp_dir):
     assert filenames[1].endswith('test_4.txt')
 
 
+def test_logging_config(temp_dir):
+    assert len(os.listdir(temp_dir)) == 0
+
+    tfut.configure_logging(temp_dir, pd.datetime.utcnow())
+    logging.getLogger().info('foobar')
+
+    assert len(os.listdir(temp_dir)) == 1
diff --git a/tests/utils_test.py b/tests/utils_test.py
@@ -0,0 +1,8 @@
+from trufflepig.utils import progressbar
+
+
+def test_progressbar():
+    result = []
+    for irun in range(100):
+        result.append(progressbar(irun, 100, percentage_step=1))
+    assert all(result)