forcing num arg of numpy.linspace to be integer

KarrLab · Jan 8, 2020 · 785595d · 785595d
1 parent 127046a
commit 785595d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_run_results.py b/tests/test_run_results.py
@@ -89,7 +89,7 @@ def test_get(self):
         for component in RunResults.COMPONENTS:
             self.assertTrue(self.run_results_1_cmpt.get(component).equals(run_results_2.get(component)))
 
-        expected_times = pandas.Float64Index(numpy.linspace(0, self.max_time, 1 + self.max_time/self.checkpoint_period))
+        expected_times = pandas.Float64Index(numpy.linspace(0, self.max_time, int(1 + self.max_time/self.checkpoint_period)))
         for component in ['populations', 'observables', 'functions', 'aggregate_states', 'random_states']:
             component_data = self.run_results_1_cmpt.get(component)
             self.assertFalse(component_data.empty)

diff --git a/wc_sim/testing/verify.py b/wc_sim/testing/verify.py
@@ -139,7 +139,7 @@ def read_expected_predictions(self):
             self.test_case_dir, self.test_case_num+'-results.csv')
         expected_predictions_df = pd.read_csv(expected_predictions_file)
         # expected predictions should contain data for all time steps
-        times = np.linspace(self.settings['start'], self.settings['duration'], num=self.settings['steps']+1)
+        times = np.linspace(self.settings['start'], self.settings['duration'], num=int(self.settings['steps'] + 1))
         if not np.allclose(times, expected_predictions_df.time):
             raise VerificationError("times in settings '{}' differ from times in expected predictions '{}'".format(
                 self.settings_file, expected_predictions_file))