diff --git a/tests/test_ps_data.py b/tests/test_ps_data.py index 08420b65f..b05d0fa04 100644 --- a/tests/test_ps_data.py +++ b/tests/test_ps_data.py @@ -85,18 +85,14 @@ def test_events_construct(self): end = max(next_year, manual_events[-1].end_dt) all_events = list(ps_data.events(end=end)) - # As there's no __eq__ for PSEvent yet - manual_start_dts = [e.start_dt for e in manual_events] - all_start_dts = [e.start_dt for e in all_events] - assert set(all_start_dts) > set(manual_start_dts) + manual_dates = [e.date for e in manual_events] + all_dates = [e.date for e in all_events] + assert set(all_dates) > set(manual_dates) def test_override(self): ps_100 = ps_data.get_ps_event_by_number(100) assert 'ONE HUNDREDTH' in ps_100.description - # For now, don't test .date, as it's either date() or datetime(), - # depending on whether the event is from the algorithm or not - def assertStartsAt(self, event, dt): assert event.start_dt == dt, '%r is wrong start time' % event.start_dt