def test_get_events_by_types(self): th = TaskHistory() th.log_unconverged() th.log_unconverged() th.log_abinit_stop(run_time=100) th.log_finalized() events = th.get_events_by_types([TaskEvent.UNCONVERGED, TaskEvent.FINALIZED]) self.assertEqual(len(th), 4) self.assertEqual(len(events), 3)