Skip to content

Commit 75a3c14

Browse files
committed
tests(dsew test_pull): fix test
1 parent 40aa6b0 commit 75a3c14

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dsew_community_profile/tests/test_pull.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -535,7 +535,10 @@ def test_interpolation(self):
535535
missing_dfs = [missing_sig1, missing_sig2, missing_sig3, missing_sig4]
536536
interpolated_dfs1 = interpolate_missing_values({("src", "sig", False): pd.concat(missing_dfs)})
537537
expected_dfs = pd.concat([sig1, sig2, sig4.loc[9:]])
538-
pd.testing.assert_frame_equal(interpolated_dfs1[("src", "sig", False)], expected_dfs)
538+
pd.testing.assert_frame_equal(
539+
interpolated_dfs1[("src", "sig", False)].sort_values(["geo_id", "timestamp"], ignore_index=True),
540+
expected_dfs.sort_values(["geo_id", "timestamp"], ignore_index=True)
541+
)
539542

540543
def test_interpolation_object_type(self):
541544
DTYPES = {"geo_id": str, "timestamp": "datetime64[ns]", "val": float, "se": float, "sample_size": float, "publish_date": "datetime64[ns]"}

0 commit comments

Comments
 (0)