test new index in inferences testing for windows os

tests/test_inferences.py

@@ -105,7 +105,10 @@ def stddev(self):
         niter=niter
     )
 
-    expected_index = np.arange(6)
+    # by forcing the expected_index to be of type RangeIndex we avoid mismatching
+    # types on Windows OS.
+    expected_index = pd.RangeIndex(start=0, stop=6, step=1)
+
     # test complete_preds_means
     expec_complete_preds_means = pd.DataFrame(
         data=np.array([7, 7, 7, 16, 16, 16]),
@@ -115,8 +118,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_complete_preds_means['complete_preds_means'],
-        inferences['complete_preds_means'],
-        check_dtype=False,
+        inferences['complete_preds_means']
     )
     simulated_pre_ys = maybe_unstandardize(
         np.squeeze(one_step_dist.sample(niter)),
@@ -147,8 +149,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_complete_preds_lower['complete_preds_lower'],
-        inferences['complete_preds_lower'],
-        check_dtype=False,
+        inferences['complete_preds_lower']
     )
     # test complete_preds_upper
     expec_complete_preds_upper = np.concatenate([pre_preds_upper, post_preds_upper])
@@ -171,8 +172,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_preds_means['post_preds_means'],
-        inferences['post_preds_means'],
-        check_dtype=False,
+        inferences['post_preds_means']
     )
     # test post_preds_lower
     expec_post_preds_lower = pd.DataFrame(
@@ -183,8 +183,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_preds_lower['post_preds_lower'],
-        inferences['post_preds_lower'],
-        check_dtype=False,
+        inferences['post_preds_lower']
     )
     # test post_preds_upper
     expec_post_preds_upper = pd.DataFrame(
@@ -195,8 +194,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_preds_upper['post_preds_upper'],
-        inferences['post_preds_upper'],
-        check_dtype=False,
+        inferences['post_preds_upper']
     )
     # test post_cum_Y
     post_cum_y = np.concatenate([[np.nan] * (len(pre_data) - 1) + [0],
@@ -209,8 +207,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_cum_y['post_cum_y'],
-        inferences['post_cum_y'],
-        check_dtype=False,
+        inferences['post_cum_y']
     )
     # test post_cum_preds_means
     expec_post_cum_preds_means = np.cumsum(expec_post_preds_means)
@@ -221,8 +218,7 @@ def stddev(self):
     expec_post_cum_preds_means['post_cum_preds_means'][len(pre_data) - 1] = 0
     pd.testing.assert_series_equal(
         expec_post_cum_preds_means['post_cum_preds_means'],
-        inferences['post_cum_preds_means'],
-        check_dtype=False,
+        inferences['post_cum_preds_means']
     )
     # test post_cum_preds_lower
     post_cum_preds_lower, post_cum_preds_upper = np.percentile(
@@ -242,8 +238,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_cum_preds_lower['post_cum_preds_lower'],
-        inferences['post_cum_preds_lower'],
-        check_dtype=False,
+        inferences['post_cum_preds_lower']
     )
     # test post_cum_preds_upper
     post_cum_preds_upper = np.concatenate(
@@ -258,8 +253,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_cum_preds_upper['post_cum_preds_upper'],
-        inferences['post_cum_preds_upper'],
-        check_dtype=False,
+        inferences['post_cum_preds_upper']
     )
     # test point_effects_means
     net_data = pd.concat([pre_data, post_data])
@@ -273,8 +267,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_point_effects_means['point_effects_means'],
-        inferences['point_effects_means'],
-        check_dtype=False,
+        inferences['point_effects_means']
     )
     # test point_effects_lower
     expec_point_effects_lower = net_data.iloc[:, 0] - inferences['complete_preds_upper']
@@ -286,8 +279,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_point_effects_lower['point_effects_lower'],
-        inferences['point_effects_lower'],
-        check_dtype=False,
+        inferences['point_effects_lower']
     )
     # test point_effects_upper
     expec_point_effects_upper = net_data.iloc[:, 0] - inferences['complete_preds_lower']
@@ -299,8 +291,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_point_effects_upper['point_effects_upper'],
-        inferences['point_effects_upper'],
-        check_dtype=False,
+        inferences['point_effects_upper']
     )
     # test post_cum_effects_means
     post_data = post_data.set_index(pd.RangeIndex(start=3, stop=6))
@@ -315,8 +306,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_cum_effects_means['post_cum_effects_means'],
-        inferences['post_cum_effects_means'],
-        check_dtype=False,
+        inferences['post_cum_effects_means']
     )
     # test post_cum_effects_lower
     post_cum_effects_lower, post_cum_effects_upper = np.percentile(
@@ -336,8 +326,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_cum_effects_lower['post_cum_effects_lower'],
-        inferences['post_cum_effects_lower'],
-        check_dtype=False,
+        inferences['post_cum_effects_lower']
     )
     # test post_cum_effects_upper
     post_cum_effects_upper = np.concatenate(
@@ -352,8 +341,7 @@ def stddev(self):
     )
     pd.testing.assert_series_equal(
         expec_post_cum_effects_upper['post_cum_effects_upper'],
-        inferences['post_cum_effects_upper'],
-        check_dtype=False,
+        inferences['post_cum_effects_upper']
     )