Ensure consistent dtypes in tests of yv_ya

iiasa · May 24, 2022 · d90a27d · d90a27d
1 parent 921ea78
commit d90a27d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/message_ix/core.py b/message_ix/core.py
@@ -606,7 +606,7 @@ def years_active(self, node: str, tec: str, yr_vtg: Union[int, str]) -> List[int
             data.where(data.age.shift(1, fill_value=0) < lt)
             .where(data.year >= yv)["year"]
             .dropna()
-            .astype(int)
+            .astype(np.int64)
             .tolist()
         )
 

diff --git a/message_ix/tests/test_feature_vintage_and_active_years.py b/message_ix/tests/test_feature_vintage_and_active_years.py
@@ -130,9 +130,7 @@ def test_vintage_and_active_years1(test_mp):
         obs = scen.vintage_and_active_years(
             ya_args=("foo", "bar", years[0]), in_horizon=True
         )
-    assert_frame_equal(
-        pd.DataFrame(columns=["year_vtg", "year_act"]), obs, check_dtype=False
-    )
+    assert_frame_equal(pd.DataFrame(columns=["year_vtg", "year_act"]), obs)
 
     # Exception is raised for incorrect arguments
     with pytest.raises(ValueError, match=r"got \('foo',\) of length 1"):