Add parameterized test for from_rfc3339 with nanos (#7675)

crwilcox · web-flow · commit ffc97ed23ca7 · 2019-05-20T12:20:32.000-07:00
* Add parameterized test for from_rfc3339 with nanos
diff --git a/api_core/tests/unit/test_datetime_helpers.py b/api_core/tests/unit/test_datetime_helpers.py
@@ -269,6 +269,24 @@ def test_from_rfc3339_w_full_precision():
         stamp = datetime_helpers.DatetimeWithNanoseconds.from_rfc3339(timestamp)
         assert stamp == expected
 
+    @staticmethod
+    @pytest.mark.parametrize(
+        "fractional, nanos",
+        [
+            ("12345678", 123456780),
+            ("1234567", 123456700),
+            ("123456", 123456000),
+            ("12345", 123450000),
+            ("1234", 123400000),
+            ("123", 123000000),
+            ("12", 120000000),
+            ("1", 100000000),
+        ],
+    )
+    def test_from_rfc3339_test_nanoseconds(fractional, nanos):
+        value = "2009-12-17T12:44:32.{}Z".format(fractional)
+        assert datetime_helpers.DatetimeWithNanoseconds.from_rfc3339(value).nanosecond == nanos
+
     @staticmethod
     def test_timestamp_pb_wo_nanos_naive():
         stamp = datetime_helpers.DatetimeWithNanoseconds(
