diff --git a/test.py b/test.py index c3b22e0..d679738 100755 --- a/test.py +++ b/test.py @@ -58,13 +58,13 @@ class TestShowAll(unittest.TestCase): class TestAdd(unittest.TestCase): def setUp(self): import shutil - shutil.rmtree(os.path.join(TEST_DATA_DIR, "test3")) - os.mkdir(os.path.join(TEST_DATA_DIR, "test3")) + shutil.rmtree(os.path.join(TEST_DATA_DIR, "test_add")) + os.mkdir(os.path.join(TEST_DATA_DIR, "test_add")) def test_add_sub(self): - with open(os.path.join(TEST_DATA_DIR, "test3/todo.txt"), "w") as f: + with open(os.path.join(TEST_DATA_DIR, "test_add/todo.txt"), "w") as f: f.write("(Z) gibberish") - setup_environment("test3") + setup_environment("test_add") sub.addSub(1, "sub", False) out = sub.getSub(sub.getItem(1)) self.assertTrue(out is not None)