Skip to content

Commit

Permalink
Tweak tests
Browse files Browse the repository at this point in the history
  • Loading branch information
marksteward committed Mar 27, 2016
1 parent 44c159a commit 4b39a58
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions tests/test_ps_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 4b39a58

Please sign in to comment.