class TestDataRegistry(unittest.TestCase): def setUp(self): self.base_temp_dir = tempfile.mkdtemp() self.dirs = [ ("TEST1", os.path.join(self.base_temp_dir, "test1/"), None), ("TEST2", "test2/", "TEST1"), ] self.files = [ ("ROOTFILE", "root.txt", None), ("TEST1_FILE", "test1file.txt", "TEST1"), ("TEST2_FILE", "test2file.txt", "TEST2"), ] self.data_reg = DataRegistry(dirs=self.dirs, files=self.files) pass def tearDown(self): shutil.rmtree(self.base_temp_dir) del self.data_reg pass def test_exceptions(self): with self.assertRaises(NotRegistered): self.data_reg.get_directory_path("TEST3") with self.assertRaises(AlreadyRegistered): self.data_reg.register_data_directory("TEST1", "false_test1/", None) def test_iter(self): for path in self.data_reg.get_all_files(): self.assertNotIn(path, [None, ""]) pass # end of class