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")
