diff --git a/wizard-work/beam/test_lcs.py b/wizard-work/beam/test_lcs.py index 92924ea..9fb30f7 100644 --- a/wizard-work/beam/test_lcs.py +++ b/wizard-work/beam/test_lcs.py @@ -1,5 +1,5 @@ -import sys -sys.path.insert(0, "/root/.hermes/wizard-outputs/beam") +import os, sys +sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) from lcs import lcs_length, lcs, lcs_diff diff --git a/wizard-work/beam/test_merge_intervals.py b/wizard-work/beam/test_merge_intervals.py index 260bdfd..661abf1 100644 --- a/wizard-work/beam/test_merge_intervals.py +++ b/wizard-work/beam/test_merge_intervals.py @@ -30,4 +30,8 @@ def test_dup(): if __name__=="__main__": for n in [x for x in globals() if x.startswith("test_")]: globals()[n]() - print("ALL TESTS PASSED") \ No newline at end of file + print("ALL TESTS PASSED") + +# Run at import time so unittest discover exercises these cases. +for n in [x for x in globals() if x.startswith("test_")]: + globals()[n]() \ No newline at end of file diff --git a/wizard-work/beam/toposort/test_toposort.py b/wizard-work/beam/toposort/test_toposort.py index 8981fbc..a675176 100644 --- a/wizard-work/beam/toposort/test_toposort.py +++ b/wizard-work/beam/toposort/test_toposort.py @@ -1,5 +1,5 @@ -import sys -sys.path.insert(0, "/root/.hermes/wizard-outputs/beam") +import os, sys +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from toposort import topo_sort def check(name, cond):