Add support for a working dir in wizard.shell

[wizard.git] / wizard / tests / scripts_test.py

diff --git a/wizard/tests/scripts_test.py b/wizard/tests/scripts_test.py
index d5a7abd1c300bbf3bc82acc90d92038a34c91d10..9cb120e1ebbd38e9d64a297abca77bf1b3f10fef 100644 (file)
--- a/wizard/tests/scripts_test.py
+++ b/wizard/tests/scripts_test.py
@@ -1,7 +1,9 @@
-from wizard.scripts import *
 import os
+import unittest
 
-testDirectory = os.path.dirname(os.path.abspath(__file__))
+from wizard import tests
+from wizard.scripts import *
 
-def test_get_disk_usage():
-    assert get_disk_usage(os.path.join(testDirectory, "scripts_test/"), "ignore_me") == 7
+class GetDiskUsageTest(unittest.TestCase):
+    def basicTest(self):
+        self.assertEqual(get_disk_usage(tests.getTestFile("scripts_test"), "ignore_me"), 7)