Fixing tests with expanded timespan

ciemss · Apr 17, 2024 · d24d9d9 · d24d9d9
1 parent 927318b
commit d24d9d9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/pyciemss/interfaces.py b/pyciemss/interfaces.py
@@ -145,10 +145,7 @@ def ensemble_sample(
         model = EnsembleCompiledDynamics.load(
             model_paths_or_jsons, dirichlet_alpha, solution_mappings
         )
-
-        # logging_times = torch.arange(
-        #     start_time + logging_step_size, end_time, logging_step_size
-        # )
+
         logging_times = torch.arange(
             start_time, end_time + logging_step_size, logging_step_size
         )

diff --git a/tests/test_interfaces.py b/tests/test_interfaces.py
@@ -539,7 +539,7 @@ def test_output_format(
     )["data"]
     assert isinstance(processed_result, pd.DataFrame)
     assert processed_result.shape[0] == num_samples * len(
-        torch.arange(start_time + logging_step_size, end_time, logging_step_size)
+        torch.arange(start_time, end_time + logging_step_size, logging_step_size)
     )
     assert processed_result.shape[1] >= 2
     assert list(processed_result.columns)[:3] == [

diff --git a/tests/visuals/test_plots.py b/tests/visuals/test_plots.py
@@ -51,7 +51,7 @@ def create_distributions(logging_step_size=20, time_unit="twenty"):
         sample,
         # using same time point formula as in 'logging_times' from  pyciemms interfaces formula 'sample'
         timepoints=np.arange(
-            start_time + logging_step_size, end_time, logging_step_size
+            start_time, end_time + logging_step_size, logging_step_size
         ),
         time_unit=time_unit,
     )[0]