Merge remote-tracking branch 'origin/main' into databricks-compatibility

tdspora · Oct 17, 2024 · cb852d7 · cb852d7
2 parents 5682545 + 342775c
commit cb852d7
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/src/tests/unit/dataset/test_dataset.py b/src/tests/unit/dataset/test_dataset.py
@@ -458,6 +458,24 @@ def test_define_date_format_with_diff_format(
             "14-11-2074", "16-06-2089", "02-11-2082", "23-08-2082",
             "28-12-2072", "02-04-2076", "03-10-2086", "18-10-2071"
         ]),
+        ("%d-%m-%Y", "%d-%m-%Y", [
+            "01-01-2074", "27-03-2081", "17-03-2087", "09-09-2089",
+            "09-08-2086", "11-04-2082", "12-04-2072", "06-09-2081",
+            "09-09-2080", "23-03-2070", "17-09-2070", "06-03-2082",
+            "23-03-2078", "24-03-2076", "31-03-2073", "28-10-2084",
+            "28-09-2074", "27-04-2073", "15-03-2086", "21-10-2080",
+            "08-04-2087", "08-07-2070", "13-03-2073", "21-10-2071",
+            "17-04-2076", "27-12-2081", "08-04-2070", "01-03-2076",
+            "13-02-2071", "05-04-2082", "30-09-2085", "02-04-2071",
+            "10-10-2084", "31-10-2085", "16-04-2080", "02-10-2070",
+            "26-07-2073", "12-03-2073", "02-03-2074", "16-09-2077",
+            "24-03-2070", "12-02-2088", "20-06-2070", "08-08-2075",
+            "13-11-2087", "30-11-2086", "26-07-2084", "26-06-2081",
+            "22-12-2075", "15-07-2079", "23-09-2070", "23-09-2077",
+            "14-11-2074", "16-06-2089", "02-11-2082", "23-08-2082",
+            "28-12-2072", "02-04-2076", "03-10-2086", "18-10-2071",
+            np.nan, np.nan, np.nan
+        ]),
         ("%m/%d/%Y", "%m/%d/%Y", [
             "02/27/2089", "11/24/2088", "02/16/2087", "08/30/2071",
             "12/14/2071", "09/05/2074", "05/28/2081", "04/24/2075",
@@ -550,8 +568,7 @@ def test_define_date_format_with_diff_format_and_provided_data(
     [
         ("%Y-%m-%d", "%Y-%m-%d", ["4723-10-17T07:45:35Z", "9999-12-31T05:22:15Z"]),
         ("%Y/%m/%d", "%Y/%m/%d", ["3/10/17T07:45:35Z", "9/12/31T05:22:15Z"]),
-        ("%m/%d/%Y", "%m/%d/%Y", ["11/30/2017T07:45:35Z", "02/27/1999T05:22:15Z"]),
-        ("%d-%m-%Y", "%d-%m-%Y", [np.nan, np.nan, np.nan])
+        ("%m/%d/%Y", "%m/%d/%Y", ["11/30/2017T07:45:35Z", "02/27/1999T05:22:15Z"])
     ]
 )
 @patch("syngen.ml.vae.models.dataset.fetch_config", return_value=MagicMock())