def test_interval_is_1(self, mock_save_check):
state = {}
check = Interval('test_file', period=1)
check.on_end_epoch(state)
check.on_end_epoch(state)
self.assertTrue(mock_save_check.call_count == 2)
def test_interval_is_1(self, mock_save_check):
state = {}
check = Interval('test_file', period=1)
check.on_end_epoch(state)
check.on_end_epoch(state)
self.assertTrue(mock_save_check.call_count == 2)
def test_interval_is_more_than_1(self, mock_save_check):
state = {}
check = Interval('test_file', period=4)
for i in range(13):
check.on_end_epoch(state)
if i == 3:
self.assertTrue(mock_save_check.call_count == 1)
elif i == 6:
self.assertFalse(mock_save_check.call_count == 2)
elif i == 7:
self.assertTrue(mock_save_check.call_count == 2)
self.assertTrue(mock_save_check.call_count == 3)
def test_interval_is_more_than_1(self, mock_save_check):
state = {}
check = Interval('test_file', period=4)
for i in range(13):
check.on_end_epoch(state)
if i == 3:
self.assertTrue(mock_save_check.call_count == 1)
elif i == 6:
self.assertFalse(mock_save_check.call_count == 2)
elif i == 7:
self.assertTrue(mock_save_check.call_count == 2)
self.assertTrue(mock_save_check.call_count == 3)
