import pytest
from datetime import date, datetime


@pytest.mark.parametrize("a_date,count", [
    # Year
    (2018, 3), ((2018,), 3),
    # Month
    ((2018, 1), 3),
    # Day
    ((2018, 1, 11), 1), (date(2018, 1, 11), 1), ("20180111", 1),
    ((2018, 1, 9), 0), (date(2018, 1, 9), 0), ("20180109", 0),
    # Datetime
    (datetime(2018, 1, 11, 10, 0, 0), 1),
    (datetime(2018, 1, 9, 10, 0, 0), 0)
])
def test_at_input_arguments(a_date, count, calendars):
    events = calendars.duration.at(a_date)
    assert len(events) == count