def test_moving_up_down(self):
dp = DatePicker(get_mock_input(), get_mock_output(), name=dp_name)
dp.refresh = lambda *args, **kwargs: None
dp._set_month_year(7, 2018)
dp.set_current_day(1)
dp.move_down()
assert (dp.get_current_day() == 8)
dp.move_up()
assert (dp.get_current_day() == 1)
def test_moving_right_left(self):
dp = DatePicker(get_mock_input(), get_mock_output(), name=dp_name)
dp.refresh = lambda *args, **kwargs: None
dp._set_month_year(7, 2018)
dp.set_current_day(1)
dp.move_right()
assert (dp.get_current_day() == 2)
assert (dp.selected_option == {"x": 0, "y": 1})
dp.set_current_day(1)
assert (dp.selected_option == {"x": 6, "y": 0})
dp.move_left()
assert (dp.current_month == 6)
assert (dp.current_year == 2018)
assert (dp.selected_option == {"x": 5, "y": 4})