From 2b7cccc4d9c834a722009f848fdcd132b9019817 Mon Sep 17 00:00:00 2001 From: tjlane Date: Sun, 3 Nov 2024 22:23:04 +0000 Subject: [PATCH] platform safe type comparison --- test/unit/test_rsmap.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/unit/test_rsmap.py b/test/unit/test_rsmap.py index b96a519..7de9bac 100644 --- a/test/unit/test_rsmap.py +++ b/test/unit/test_rsmap.py @@ -97,14 +97,14 @@ def test_filter_common_indices_with_maps(noise_free_map: Map) -> None: def test_verify_type(noise_free_map: Map) -> None: dataseries = rs.DataSeries([1, 2, 3]) - assert dataseries.dtype == int + assert np.issubdtype(dataseries.dtype, np.integer) noise_free_map._verify_type("foo", [int], dataseries, fix=False, cast_fix_to=int) with pytest.raises(AssertionError): noise_free_map._verify_type("foo", [float], dataseries, fix=False, cast_fix_to=float) output = noise_free_map._verify_type("foo", [float], dataseries, fix=True, cast_fix_to=float) - assert output.dtype == float + assert np.issubdtype(output.dtype, np.floating) @pytest.mark.parametrize("fix", [False, True])