class TestPlotEntities(TestCase):
def setUp(self):
self.plot = Plot("star wars")
self.plot.add_entity("vader")
self.plot.add_entity("luke")
self.plot.add_entity("emperor")
def test_plot_entites(self):
self.assertTrue(self.plot.has_entity("vader"))
self.assertTrue(self.plot.entity_alive("vader", 1.0))
self.plot.edit_entity("vader", end=0.9)
self.assertTrue(self.plot.entity_alive("vader", 0.89))
self.assertFalse(self.plot.entity_alive("vader", 0.9))
self.assertFalse(self.plot.entity_alive("vader", 0.91))
def test_plot_events(self):
self.plot.add_event("duel", 0.9, ["vader", "luke", "emperor"])
self.plot.add_event("hand loss", 0.5, ["vader", "luke"])
self.assertTrue(len(self.plot.events) == 2)
self.assertTrue(self.plot.get_sorted_events()[0].name == "hand loss")
