def test_context_nested(self):
p = Progress(num_steps=2)
self.assertFalse(p.is_started())
self.assertFalse(p.is_completed())
self.assertEqual(len(p.progress_history), 1)
self.assertEqual(p.completion(), 0.0)
with p:
p.update(msg="do a thing")
self.assertEqual(len(p.progress_history), 2)
for i in range(5):
with p:
p.update(
msg="do a thing in a loop nested in the other thing")
self.assertEqual(len(p.progress_history), 3 + i)
self.assertFalse(p.is_completed())
self.assertEqual(len(p.progress_history), 7)
self.assertFalse(p.is_completed())
self.assertTrue(p.is_completed())
self.assertEqual(len(p.progress_history), 8)
self.assertEqual(p.completion(), 1.0)
def test_context(self):
p = Progress(num_steps=2)
self.assertFalse(p.is_started())
self.assertFalse(p.is_completed())
self.assertEqual(len(p.progress_history), 1)
self.assertEqual(p.completion(), 0.0)
with p:
p.update(msg="do a thing")
self.assertTrue(p.is_started())
self.assertEqual(len(p.progress_history), 2)
self.assertEqual(p.completion(), 0.5)
self.assertTrue(p.is_completed())
self.assertEqual(len(p.progress_history), 3)
self.assertEqual(p.completion(), 1.0)
def test_basic_single_step(self):
p = Progress(num_steps=2)
self.assertFalse(p.is_started())
self.assertFalse(p.is_completed())
self.assertEqual(len(p.progress_history), 1)
self.assertEqual(p.completion(), 0.0)
p.update(msg="do a thing")
self.assertTrue(p.is_started())
self.assertEqual(len(p.progress_history), 2)
self.assertEqual(p.completion(), 0.5)
p.mark_complete()
self.assertTrue(p.is_completed())
self.assertEqual(len(p.progress_history), 3)
self.assertEqual(p.completion(), 1.0)