test: Update tests to properly execute all code.

tektronix · Mar 4, 2024 · e3f01b2 · e3f01b2
1 parent 1d6f417
commit e3f01b2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/tests/samples/tsp_script.py → tests/samples/tsp_script.tsp b/tests/samples/tsp_script.py → tests/samples/tsp_script.tsp
@@ -1,3 +1,2 @@
 """Sample script used in test_smu.py."""
-
 print("TEK")
diff --git a/tests/test_smu.py b/tests/test_smu.py
@@ -49,7 +49,7 @@ def test_smu(  # noqa: PLR0915
         assert "Query" in stdout
 
     smu.load_script(
-        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.py",
+        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.tsp",
         run_script=True,
         to_nv_memory=True,
     )
@@ -61,7 +61,7 @@ def test_smu(  # noqa: PLR0915
     assert "loadfuncs()" in stdout
     smu.expect_esr(0)
     smu.load_script(
-        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.py",
+        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.tsp",
         script_name="tsp_function",
     )
     stdout = capsys.readouterr().out
Original file line number	Diff line number	Diff line change
		@@ -1,3 +1,2 @@
		"""Sample script used in test_smu.py."""

		print("TEK")