def test_is_tz_awareness_with_various_objects_returns_false(self): # Arrange time_object1 = datetime(1970, 1, 1, 0, 0, 0, 0) time_object2 = pd.Timestamp(datetime(1970, 1, 1, 0, 0, 0, 0)) # Act, Assert assert is_tz_aware(time_object1) is False assert is_tz_aware(time_object2) is False assert is_tz_naive(time_object1) is True assert is_tz_naive(time_object2) is True
def test_is_tz_awareness_with_various_objects_returns_false(self): # Arrange time_object1 = datetime(1970, 1, 1, 0, 0, 0, 0) time_object2 = pd.Timestamp(datetime(1970, 1, 1, 0, 0, 0, 0)) # Act # Assert self.assertFalse(is_tz_aware(time_object1)) self.assertFalse(is_tz_aware(time_object2)) self.assertTrue(is_tz_naive(time_object1)) self.assertTrue(is_tz_naive(time_object2))
def test_is_tz_awareness_with_various_aware_objects_returns_true(self): # Arrange time_object1 = UNIX_EPOCH time_object2 = pd.Timestamp(UNIX_EPOCH) time_object3 = pd.DataFrame({"timestamp": ["2019-05-21T12:00:00+00:00", "2019-05-21T12:15:00+00:00"]}) time_object3.set_index("timestamp") time_object3.index = pd.to_datetime(time_object3.index) # Act # Assert self.assertTrue(is_tz_aware(time_object1)) self.assertTrue(is_tz_aware(time_object2)) self.assertTrue(is_tz_aware(time_object3)) self.assertFalse(is_tz_naive(time_object1)) self.assertFalse(is_tz_naive(time_object2)) self.assertFalse(is_tz_naive(time_object3))
def test_is_tz_awareness_with_various_aware_objects_returns_true(self): # Arrange time_object1 = datetime(1970, 1, 1, 0, 0, 0, 0, pytz.UTC) time_object2 = pd.Timestamp(datetime(1970, 1, 1, 0, 0, 0, 0, pytz.UTC)) time_object3 = pd.DataFrame({ 'timestamp': ['2019-05-21T12:00:00+00:00', '2019-05-21T12:15:00+00:00'] }) time_object3.set_index('timestamp') time_object3.index = pd.to_datetime(time_object3.index) # Act # Assert self.assertTrue(is_tz_aware(time_object1)) self.assertTrue(is_tz_aware(time_object2)) self.assertTrue(is_tz_aware(time_object3)) self.assertFalse(is_tz_naive(time_object1)) self.assertFalse(is_tz_naive(time_object2)) self.assertFalse(is_tz_naive(time_object3))
def test_is_tz_awareness_given_unrecognized_type_raises_exception(self): # Arrange # Act # Assert with pytest.raises(ValueError): is_tz_aware("hello")