unti test for dassert_is_valid_timestamp

helpers/test/test_hdatetime.py

@@ -773,3 +773,57 @@ def test2(self) -> None:
             hdateti.dassert_str_is_date(date)
         actual = str(err.exception)
         self.check_string(actual)
+
+# #############################################################################
+# Test_dassert_is_valid_timestamp
+# #############################################################################
+
+class Test_dassert_is_valid_timestamp(hunitest.TestCase):
+
+    def test1(self) -> None:
+        """
+        Test by checking a timestamp with a timezone.
+        """
+        timestamp = pd.Timestamp("2021-01-04 09:30:00-05:00", tz="America/New_York")
+        # This should not raise any exceptions.
+        hdateti.dassert_is_valid_timestamp(timestamp)
+
+
+    def test2(self) -> None:
+        """
+        Check timestamp without a timezone info, should raise an exception.
+        """
+        timestamp = pd.Timestamp("2021-01-04 09:30:00")
+        with self.assertRaises(AssertionError) as cm:
+            hdateti.dassert_is_valid_timestamp(timestamp)
+        act = str(cm.exception)
+        exp = """
+        * Failed assertion *
+        'None' is not 'None'
+        datetime_='2021-01-04 09:30:00' doesn't have timezone info
+        """
+        #check.
+        self.assert_equal(act, exp, fuzzy_match=True)
+
+    def test3(self) -> None:
+        """
+        Test by checking a timestamp as None.
+        """
+        timestamp = None
+        # This should not raise any exceptions.
+        hdateti.dassert_is_valid_timestamp(timestamp)
+
+    def test4(self) -> None:
+        """
+        Test by passing a timestamp as a string, should raise an exception.
+        """
+        timestamp = "2021-01-04 09:30:00"
+        with self.assertRaises(AssertionError) as cm:
+            hdateti.dassert_is_valid_timestamp(timestamp)
+        act = str(cm.exception)
+        exp = """
+        * Failed assertion *
+        Instance of '2021-01-04 09:30:00' is '<class 'str'>' instead of '<class 'pandas._libs.tslibs.timestamps.Timestamp'>'
+        """
+        #check.
+        self.assert_equal(act, exp, fuzzy_match=True)