Skip to content

Commit 534207d

Browse files
authored
Fix: unit testing comparison of datetimes US/NS (#1582)
1 parent 4417614 commit 534207d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sqlmesh/core/test/definition.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,9 @@ def tearDown(self) -> None:
9494

9595
def assert_equal(self, expected: pd.DataFrame, actual: pd.DataFrame) -> None:
9696
"""Compare two DataFrames"""
97-
expected = expected.astype(actual.dtypes.to_dict())
97+
actual_types = actual.dtypes.to_dict()
98+
# two astypes are necessary, pandas converts strings to times as NS, but if the actual is US, it doesn't take affect until the 2nd try!
99+
expected = expected.astype(actual_types).astype(actual_types)
98100
expected = expected.replace({np.nan: None, "nan": None})
99101
actual = actual.replace({np.nan: None, "nan": None})
100102

0 commit comments

Comments
 (0)