From 3f0dcbdb8b4886423c50eb0faf65f1fffb87dd63db01e044bbfaaab1652cb95a Mon Sep 17 00:00:00 2001
From: Sergey Matveev <stargrave@stargrave.org>
Date: Sat, 30 Nov 2024 21:58:02 +0300
Subject: [PATCH] Replace hard-coded tests with symmetric ones

---
 pyac/tests/test_tai.py | 24 ++++++++----------------
 1 file changed, 8 insertions(+), 16 deletions(-)

diff --git a/pyac/tests/test_tai.py b/pyac/tests/test_tai.py
index 2a83f21..d286607 100644
--- a/pyac/tests/test_tai.py
+++ b/pyac/tests/test_tai.py
@@ -2,6 +2,7 @@ from datetime import datetime
 from unittest import TestCase
 
 from hypothesis import given
+from hypothesis.strategies import datetimes
 
 from pyac import DecodeError
 from pyac import dumps
@@ -50,27 +51,18 @@ DJB_Leapsecs = (
 
 
 class TestTAI64(TestCase):
-    @given(junk_st)
-    def test_encode_decode_tai64(self, junk: bytes) -> None:
-        dt = datetime(2023, 10, 1, 12, 0, 0)
+    @given(datetimes(), junk_st)
+    def test_symmetric_tai64(self, dt: datetime, junk: bytes) -> None:
+        dt = dt.replace(microsecond=0)
         encoded = dumps(dt)
-        self.assertSequenceEqual(
-            encoded,
-            b"\x18\x40\x00\x00\x00\x65\x19\x5f\x65",
-        )
         decoded, tail = loads(encoded + junk)
         self.assertEqual(dt, decoded)
         self.assertSequenceEqual(tail, junk)
 
-    @given(junk_st)
-    def test_encode_decode_tai64n(self, junk: bytes) -> None:
-        dt = datetime(2023, 10, 1, 12, 0, 0, 123456)
-        encoded = dumps(dt) + junk
-        self.assertSequenceEqual(
-            encoded,
-            b"\x19\x40\x00\x00\x00\x65\x19\x5f\x65\x07\x5b\xca\x00" + junk,
-        )
-        decoded, tail = loads(encoded)
+    @given(datetimes(), junk_st)
+    def test_symmetric_tai64n(self, dt: datetime, junk: bytes) -> None:
+        encoded = dumps(dt)
+        decoded, tail = loads(encoded + junk)
         self.assertEqual(dt, decoded)
         self.assertSequenceEqual(tail, junk)
 
-- 
2.50.0