fix tests, less iterations

Author: elazarg

experiment/trivial/instrumented.py

@@ -6,7 +6,7 @@ def run() -> None:
     with persist.Loader(__file__, locals()) as transaction:
         if transaction:
             [i] = transaction.move()
-        for i in transaction.iterate(range(1000)):  # type: int
+        for i in transaction.iterate(range(100)):  # type: int
             pass
             transaction.commit(i)
 

experiment/trivial/main.py

@@ -1,6 +1,6 @@
 def run() -> None:
     """Trivial baseline"""
-    for i in range(1000):  # type: int
+    for i in range(100):  # type: int
         pass
 
 

experiment/trivial/naive.py

@@ -3,10 +3,11 @@
 
 def run() -> None:
     """Trivial baseline"""
+    [i] = (None,) * 1
     with persist.Loader(__file__, locals()) as transaction:
         if transaction:
             [i] = transaction.move()
-        for i in transaction.iterate(range(1000)):  # type: int
+        for i in transaction.iterate(range(100)):  # type: int
             pass
             transaction.commit(i)
 

experiment/trivial/vm.py

@@ -4,7 +4,7 @@
 def run() -> None:
     """Trivial baseline"""
     with persist.SimpleTcpClient("trivial") as client:
-        for i in client.iterate(range(1000)):  # type: int
+        for i in client.iterate(range(100)):  # type: int
             pass
             client.commit()
 

tests/test_generate.py

@@ -18,8 +18,8 @@ def compare_transformed_files(actual: str, expected_outfile: pathlib.Path) -> No
     expected = expected_outfile.read_text(encoding="utf-8")
     expected_transaction = find_transaction(expected)
     actual_transaction = find_transaction(actual)
-    assert actual_transaction == expected_transaction
     assert actual == expected
+    assert actual_transaction == expected_transaction
 
 
 def naive_transform(filename: pathlib.Path, expected_outfile: pathlib.Path) -> None:

tests/test_instrument.py

@@ -1,13 +1,13 @@
+import pathlib
+
 from checkpoint import persist
 
 
 def instrument_experiment(experiment_name: str, args: list[str]) -> None:
     def run(fuel: int) -> str:
+        instrumented = pathlib.Path("experiment") / experiment_name / "instrumented.py"
         return persist.run_instrumented_file(
-            instrumented=f"experiment/{experiment_name}/instrumented.py",
-            args=args,
-            fuel=fuel,
-            capture_stdout=True,
+            instrumented=instrumented, args=args, fuel=fuel, capture_stdout=True
         )
 
     expected = run(10**6)