Test fixes

SURGroup · Apr 5, 2022 · 9e37f6f · 9e37f6f
1 parent f2dc355
commit 9e37f6f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/tests/unit_tests/inference/pfn2.py → tests/unit_tests/inference/pfn_cubic.py b/tests/unit_tests/inference/pfn2.py → tests/unit_tests/inference/pfn_cubic.py
diff --git a/tests/unit_tests/inference/pfn.py → tests/unit_tests/inference/pfn_linear.py b/tests/unit_tests/inference/pfn.py → tests/unit_tests/inference/pfn_linear.py
diff --git a/tests/unit_tests/inference/pfn1.py → tests/unit_tests/inference/pfn_quadratic.py b/tests/unit_tests/inference/pfn1.py → tests/unit_tests/inference/pfn_quadratic.py
diff --git a/tests/unit_tests/inference/test_bayes_model_selection.py b/tests/unit_tests/inference/test_bayes_model_selection.py
@@ -19,16 +19,18 @@
 
 
 def test_models():
-    os.chdir(dir_path)
+    a = os.getcwd()
+    if "inference" not in a:
+        raise NotImplementedError(a)
     data_ex1 = np.loadtxt('data_ex1a.txt')
 
-    model = PythonModel(model_script='pfn.py', model_object_name='model_linear', var_names=['theta_0'])
+    model = PythonModel(model_script='pfn_linear.py', model_object_name='model_linear', var_names=['theta_0'])
     runmodel4 = RunModel_New(model=model)
 
-    model1 = PythonModel(model_script='pfn1.py', model_object_name='model_quadratic', var_names=['theta_0', 'theta_1'])
+    model1 = PythonModel(model_script='pfn_quadratic.py', model_object_name='model_quadratic', var_names=['theta_0', 'theta_1'])
     runmodel5 = RunModel_New(model=model1)
 
-    model2 = PythonModel(model_script='pfn2.py', model_object_name='model_cubic',
+    model2 = PythonModel(model_script='pfn_cubic.py', model_object_name='model_cubic',
                          var_names=['theta_0', 'theta_1', 'theta_2'])
     runmodel6 = RunModel_New(model=model2)