test of proper directory structure creation

knstmrd · Jul 5, 2018 · ebe38d6 · ebe38d6
1 parent ed9deb7
commit ebe38d6
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/test/test_dataprocessor.py b/test/test_dataprocessor.py
@@ -0,0 +1,25 @@
+from os.path import isfile, isdir
+from shutil import rmtree
+from os import getcwd
+import unittest
+from src import dataprocessor
+
+
+class DataProcessorTest(unittest.TestCase):
+    def setUp(self):
+        self.curr_dir = getcwd() + '/'
+        self.dataprocessor = dataprocessor.DataProcessor(self.curr_dir)
+
+    def tearDown(self):
+        rmtree(self.curr_dir + 'dataprocessor_files/')
+
+    def test_directory_creation(self):
+        self.assertEqual(isdir(self.curr_dir + 'dataprocessor_files/features/correlated'), True)
+        self.assertEqual(isdir(self.curr_dir + 'dataprocessor_files/features/unimportant'), True)
+        self.assertEqual(isdir(self.curr_dir + 'dataprocessor_files/features/lists'), True)
+
+        self.assertEqual(isdir(self.curr_dir + 'dataprocessor_files/logs/cv'), True)
+        self.assertEqual(isdir(self.curr_dir + 'dataprocessor_files/logs/predictions'), True)
+
+        self.assertEqual(isdir(self.curr_dir + 'dataprocessor_files/settings'), True)
+