def test_trigger_rerun(self):
update = CompositeCheckpoint(DataCheckpoint, 2)
update.append(['abc','def'])
self._parent.trigger_update({'no-parent':update})
self._transition._do_transition.reset_mock()
# Note: We do not trigger on parent, since that would clear data.
self._transition.trigger_rerun()
self._verify_composite_result()
def test_trigger_rerun_upstream(self):
update = CompositeCheckpoint(DataCheckpoint, 2)
update.append(['abc','def'])
self._parent.trigger_update({'no-parent':update})
self._transition._do_transition.reset_mock()
# Parent has no data, rerun will clear output data, and thus also downstream.
self._parent.trigger_rerun()
# No data in parent, so this should not be called.
self._transition._do_transition.assert_not_called()
def setUp(self):
self._checkpoint = CompositeCheckpoint()
cp1 = DataCheckpoint()
cp1.append('abc')
cp2 = DataCheckpoint()
cp2.append('def')
self._checkpoint.add_checkpoint(cp1)
self._checkpoint.add_checkpoint(cp2)
class TestCompositeCheckpoint(unittest.TestCase):
def setUp(self):
self._checkpoint = CompositeCheckpoint()
cp1 = DataCheckpoint()
cp1.append('abc')
cp2 = DataCheckpoint()
cp2.append('def')
self._checkpoint.add_checkpoint(cp1)
self._checkpoint.add_checkpoint(cp2)
def test_iter(self):
a = [ i for i in self._checkpoint ]
self.assertEqual(len(a), 2)
self.assertEqual(a[0].data, 'abc')
self.assertEqual(a[1].data, 'def')
def test_len(self):
self.assertEqual(len(self._checkpoint), 2)
def test_get_slice(self):
self.assertEqual( [ c.data for c in self._checkpoint[0:2] ], ['abc','def'])
def test_attribute_data(self):
self.assertEqual( self._checkpoint.data, ['abc','def'])
def test_clear(self):
self._checkpoint.clear()
self.assertEqual(self._checkpoint.data, [None, None])
def test_replace(self):
self.assertEqual( self._checkpoint.data, ['abc','def'])
self._checkpoint.replace(['ABC','DEF'])
self.assertEqual( self._checkpoint.data, ['ABC','DEF'])
def test_append(self):
self.assertEqual( self._checkpoint.data, ['abc','def'])
self._checkpoint.append(['ABC','DEF'])
self.assertEqual( self._checkpoint.data, ['abcABC','defDEF'])
def setUp(self):
self._checkpoint = CompositeCheckpoint()
class TestCompositeCheckpointDefaultInit(unittest.TestCase):
def setUp(self):
self._checkpoint = CompositeCheckpoint()
def test_len(self):
self.assertEqual(len(self._checkpoint), 0)
def test_get_item(self):
cp = DataCheckpoint()
cp.append('abc')
self._checkpoint.add_checkpoint(cp)
self.assertEqual(self._checkpoint[0].data, 'abc')
def test_set_item(self):
cp1 = DataCheckpoint()
cp1.append('abc')
cp2 = DataCheckpoint()
cp2.append('def')
self._checkpoint.add_checkpoint(cp1)
self.assertEqual(self._checkpoint[0].data, 'abc')
self._checkpoint[0] = cp2
self.assertEqual(self._checkpoint[0].data, 'def')
def test_del_item(self):
self._checkpoint.add_checkpoint(DataCheckpoint())
self.assertEqual(len(self._checkpoint), 1)
del self._checkpoint[0]
self.assertEqual(len(self._checkpoint), 0)
def test_del_slice(self):
self._checkpoint.add_checkpoint(DataCheckpoint())
self.assertEqual(len(self._checkpoint), 1)
del self._checkpoint[0:1]
self.assertEqual(len(self._checkpoint), 0)
def test_add_checkpoint(self):
self._checkpoint.add_checkpoint(DataCheckpoint())
self.assertEqual(len(self._checkpoint), 1)
self._checkpoint.add_checkpoint(DataCheckpoint())
self.assertEqual(len(self._checkpoint), 2)
def test_remove_checkpoint(self):
self._checkpoint.add_checkpoint(DataCheckpoint())
self._checkpoint.add_checkpoint(DataCheckpoint())
self._checkpoint.remove_checkpoint(0)
self.assertEqual(len(self._checkpoint), 1)
self._checkpoint.remove_checkpoint(0)
self.assertEqual(len(self._checkpoint), 0)
def test_trigger_update_composite(self):
update = CompositeCheckpoint(DataCheckpoint, 2)
update.append(['abc','def'])
self._parent.trigger_update({'no-parent':update})
self._verify_composite_result()
def test_trigger_update_composite(self):
update = CompositeCheckpoint(DataCheckpoint, 2)
update.append(['abc','def'])
self._transition.trigger_update({'no-parent':update})
calls = [call((update[0],)), call((update[1],))]
self._transition._do_transition.assert_has_calls(calls)
