Test FlightEndpoint with nanoseconds

EnricoMi · Aug 6, 2024 · 33e09e6 · 33e09e6
1 parent 501d3a6
commit 33e09e6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/pyarrow/tests/test_flight.py b/python/pyarrow/tests/test_flight.py
@@ -251,7 +251,7 @@ def get_flight_info(self, context, descriptor):
                 flight.FlightEndpoint(
                     b'',
                     [flight.Location.for_grpc_tcp('localhost', 5005)],
-                    pa.scalar("2023-04-05T12:34:56.789").cast(pa.timestamp("ms")),
+                    pa.scalar("2023-04-05T12:34:56.789012345").cast(pa.timestamp("ns")),
                     "endpoint app metadata"
                 ),
             ],
@@ -1151,7 +1151,7 @@ def test_flight_get_info():
         assert info.endpoints[0].app_metadata == b""
         assert info.endpoints[0].locations[0] == flight.Location('grpc://test')
         assert info.endpoints[1].expiration_time == \
-            pa.scalar("2023-04-05T12:34:56.789+00:00").cast(pa.timestamp("ns", "UTC"))
+            pa.scalar("2023-04-05T12:34:56.789012345+00:00").cast(pa.timestamp("ns", "UTC"))
         assert info.endpoints[1].app_metadata == b"endpoint app metadata"
         assert info.endpoints[1].locations[0] == \
             flight.Location.for_grpc_tcp('localhost', 5005)
@@ -1763,7 +1763,7 @@ def test_roundtrip_types():
             flight.FlightEndpoint(
                 b'',
                 [flight.Location.for_grpc_tcp('localhost', 5005)],
-                pa.scalar("2023-04-05T12:34:56.789012").cast(pa.timestamp("us")),
+                pa.scalar("2023-04-05T12:34:56.789012345").cast(pa.timestamp("ns")),
                 b'endpoint app metadata'
             ),
         ],