standardise ref_file resume tests

Signed-off-by: Jack Luar <[email protected]>

Author: luarss

tools/AutoTuner/test/ref_file_check.py

@@ -3,9 +3,6 @@
 import os
 
 cur_dir = os.path.dirname(os.path.abspath(__file__))
-src_dir = os.path.join(cur_dir, "../src/autotuner")
-orfs_dir = os.path.join(cur_dir, "../../../flow")
-os.chdir(src_dir)
 
 
 class RefFileCheck(unittest.TestCase):
@@ -15,11 +12,11 @@ class RefFileCheck(unittest.TestCase):
 
     def setUp(self):
         configs = [
-            "../../test/files/no_sdc_ref.json",
-            "../../test/files/no_fr_ref.json",
+            os.path.join(cur_dir, "../../test/files/no_sdc_ref.json"),
+            os.path.join(cur_dir, "../../test/files/no_fr_ref.json"),
         ]
         self.commands = [
-            f"python3 distributed.py"
+            f"python3 -m autotuner.distributed"
             f" --design {self.design}"
             f" --platform {self.platform}"
             f" --config {c}"
@@ -28,7 +25,6 @@ def setUp(self):
             for c in configs
         ]
 
-    # Make this a test case
     def test_files(self):
         for c in self.commands:
             out = subprocess.run(c, shell=True)

tools/AutoTuner/test/resume_check.py

@@ -6,9 +6,6 @@
 from contextlib import contextmanager
 
 cur_dir = os.path.dirname(os.path.abspath(__file__))
-src_dir = os.path.join(cur_dir, "../src/autotuner")
-orfs_dir = os.path.join(cur_dir, "../../../flow")
-os.chdir(src_dir)
 
 
 @contextmanager
@@ -34,7 +31,13 @@ class ResumeCheck(unittest.TestCase):
 
     def setUp(self):
         self.config = os.path.join(
-            orfs_dir, "designs", self.platform, self.design, "autotuner.json"
+            cur_dir,
+            "../../../",
+            "flow",
+            "designs",
+            self.platform,
+            self.design,
+            "autotuner.json",
         )
         self.jobs = self.samples
         self.num_cpus = os.cpu_count()
@@ -47,7 +50,7 @@ def setUp(self):
         res_per_trial = float("{:.1f}".format(self.num_cpus / self.samples))
         options = ["", "--resume"]
         self.commands = [
-            f"python3 distributed.py"
+            f"python3 -m autotuner.distributed"
             f" --design {self.design}"
             f" --platform {self.platform}"
             f" --config {self.config}"