revert timestamp changes

Author: rahul2393

tests/system/_sample_data.py

@@ -69,42 +69,21 @@
 
 def _assert_timestamp(value, nano_value):
     assert isinstance(value, datetime.datetime)
-    # Treat naive datetimes as UTC
-    if value.tzinfo is None:
-        value_utc = value.replace(tzinfo=UTC)
+    assert value.tzinfo is None
+    assert nano_value.tzinfo is UTC
+
+    assert value.year == nano_value.year
+    assert value.month == nano_value.month
+    assert value.day == nano_value.day
+    assert value.hour == nano_value.hour
+    assert value.minute == nano_value.minute
+    assert value.second == nano_value.second
+    assert value.microsecond == nano_value.microsecond
+
+    if isinstance(value, datetime_helpers.DatetimeWithNanoseconds):
+        assert value.nanosecond == nano_value.nanosecond
     else:
-        value_utc = value.astimezone(UTC)
-    if nano_value.tzinfo is None:
-        nano_value_utc = nano_value.replace(tzinfo=UTC)
-    else:
-        nano_value_utc = nano_value.astimezone(UTC)
-
-    # Compare timestamps with tolerance for timezone differences
-    # Allow up to 24 hours difference to handle timezone conversions and date boundaries
-    time_diff = abs((value_utc - nano_value_utc).total_seconds())
-    assert time_diff <= 86400, f"Time difference {time_diff} seconds exceeds 24 hours"
-
-    # Only compare nanoseconds if the timestamps are within 1 second
-    if time_diff < 1:
-        if isinstance(value, datetime_helpers.DatetimeWithNanoseconds):
-            expected_ns = value.nanosecond
-            found_ns = (
-                nano_value.nanosecond
-                if hasattr(nano_value, "nanosecond")
-                else nano_value.microsecond * 1000
-            )
-            # Allow up to 1 second difference for timestamp precision issues
-            # This accounts for potential precision loss during database round-trip
-            ns_diff = abs(expected_ns - found_ns)
-            assert ns_diff <= 1_000_000_000, f"Nanosecond diff {ns_diff} > 1s"
-        else:
-            # Allow up to 1 microsecond difference for timestamp precision issues
-            us_diff = abs(value.microsecond - nano_value.microsecond)
-            if us_diff > 1:
-                print(f"  Expected: {value} (microsecond: {value.microsecond})")
-                print(f"  Found: {nano_value} (microsecond: {nano_value.microsecond})")
-                print(f"  Difference: {us_diff} microseconds")
-            assert us_diff <= 1, f"Microsecond diff {us_diff} > 1"
+        assert value.microsecond * 1000 == nano_value.nanosecond
 
 
 def _check_rows_data(rows_data, expected=ROW_DATA, recurse_into_lists=True):
@@ -130,17 +109,9 @@ def _check_cell_data(found_cell, expected_cell, recurse_into_lists=True):
     elif isinstance(found_cell, float) and math.isnan(found_cell):
         assert math.isnan(expected_cell)
 
-    elif (
-        isinstance(found_cell, list)
-        and isinstance(expected_cell, list)
-        and all(isinstance(x, datetime.datetime) for x in found_cell)
-    ):
-        assert len(found_cell) == len(expected_cell)
-        for found_item, expected_item in zip(found_cell, expected_cell):
-            _assert_timestamp(expected_item, found_item)
-
     elif isinstance(found_cell, list) and recurse_into_lists:
         assert len(found_cell) == len(expected_cell)
+
         for found_item, expected_item in zip(found_cell, expected_cell):
             _check_cell_data(found_item, expected_item)