Commit df4a5645 authored by Patrick Kimber's avatar Patrick Kimber

test 'get_calendars' and add month count #1181

parent febf43b3
......@@ -42,7 +42,9 @@ def test_get_bookings():
def test_get_calendars():
_demo_data()
c = HtmlCalendar()
c.get_calendars()
result = c.get_calendars()
assert list is type(result)
assert 12 == len(result)
@pytest.mark.django_db
......@@ -54,8 +56,8 @@ def test_get_calendars_html():
)
c = HtmlCalendar()
result = c.get_calendars(count=1)
assert type(result) is list
assert len(result) == 1
assert list is type(result)
assert 1 == len(result)
html = result[0]
assert "<td>02</td>" in html
assert "<td class='afternoon'>03</td>" in html
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment