init tests for %n support in strptime

Author: jyalim

Lib/test/datetimetester.py

@@ -2220,6 +2220,19 @@ def test_strptime_t_format(self):
                     self.theclass(test_year,test_month,test_day)
                 )
 
+    def test_strptime_n_format(self):
+        test_year,test_month,test_day = 2026,2,25
+        whitespaces = ('',' ','\t','\r','\v','\n','\f')
+        for ws in (*whitespaces,''.join(whitespaces)):
+            with self.subTest(whitespace=ws):
+                self.assertEqual(
+                    self.theclass.strptime(
+                        f'{test_year:04d}{ws}{test_month:02d}{ws}{test_day:02d}',
+                        "%Y%n%m%n%d"
+                    ),
+                    self.theclass(test_year,test_month,test_day)
+                )
+
 
 #############################################################################
 # datetime tests

Lib/test/test_strptime.py

@@ -686,6 +686,22 @@ def test_strptime_t_format(self):
                     )
                 )
 
+    def test_strptime_n_format(self):
+        test_year,test_month,test_day = 2026,2,20
+        whitespaces = ('',' ','\t','\r','\v','\n','\f')
+        for ws in (*whitespaces,''.join(whitespaces)):
+            with self.subTest(whitespace=ws):
+                self.assertEqual(
+                    time.strptime(
+                        f'{test_year:04d}{ws}{test_month:02d}{ws}{test_day:02d}',
+                        "%Y%n%m%n%d"
+                    ),
+                    time.strptime(
+                        f'{test_year:04d}-{test_month:02d}-{test_day:02d}',
+                        "%Y-%m-%d"
+                    )
+                )
+
 class Strptime12AMPMTests(unittest.TestCase):
     """Test a _strptime regression in '%I %p' at 12 noon (12 PM)"""
 

Lib/test/test_time.py

@@ -359,7 +359,7 @@ def test_strptime(self):
         # raising an exception.
         tt = time.gmtime(self.t)
         for directive in ('a', 'A', 'b', 'B', 'c', 'd', 'D', 'F', 'H', 'I',
-                          'j', 'm', 'M', 'p', 'S', 't', 'T',
+                          'j', 'm', 'M', 'n', 'p', 'S', 't', 'T',
                           'U', 'w', 'W', 'x', 'X', 'y', 'Y', 'Z', '%'):
             format = '%' + directive
             if directive == 'd':