def test_get_terms(dummy_service):
ff = flippyflop.FlippyFlop(dummy_service, TEST_SPREADSHEET_ID)
result = ff.get_terms()
assert result.shape == (3, 2)
assert result.iloc[0]["back"] == "Paris"
assert result.iloc[2]["front"] == "How many legs does a horse have?"
def test_todays_cards(dummy_service, freezer, date, expected):
# assuming a ff.start_date of 2019-12-31 (and in the sheet)
with freeze_time(date, tick=True):
ff = flippyflop.FlippyFlop(dummy_service, TEST_SPREADSHEET_ID)
result = ff.todays_cards()
assert result == expected
def test_get_buckets(dummy_service):
ff = flippyflop.FlippyFlop(dummy_service, TEST_SPREADSHEET_ID)
result = ff.get_buckets()
assert result.values.tolist() == [
["1", "1577865600"],
["2", "1577865660"],
]
def test_add_term(dummy_service):
ff = flippyflop.FlippyFlop(dummy_service, TEST_SPREADSHEET_ID)
ff.add_term(
front="What should horses in training wear in the stable?",
back="A Blanket",
)
assert ff.get_terms().values[-1].tolist() == [
"What should horses in training wear in the stable?",
"A Blanket",
]
def test_update_bucket(card, success, expected, freezer, dummy_service):
ff = flippyflop.FlippyFlop(dummy_service, TEST_SPREADSHEET_ID)
ff.update_bucket(card, success)
# TODO: Refactor. After adding the throttle decorator this is messy.
new_buckets = ff.get_buckets()
result = new_buckets.reset_index().values.tolist()
result_times = [int(row[-1]) for row in result]
expected_times = [int(row[-1]) for row in expected]
for idx in range(len(result_times)):
assert result_times[idx] - expected_times[idx] < 10
assert [row[:-1] for row in result] == [row[:-1] for row in expected]
def test_functional(dummy_service, freezer):
"""a functional test where cards are tested on sequential days
We assert that the cards are placed into the correct bucket each day
as well as we are testing the correct cards on each day
"""
ff = flippyflop.FlippyFlop(dummy_service, TEST_SPREADSHEET_ID)
first_test_date = datetime.datetime(2020, 1, 2, 8)
for day in range(len(tests[0])):
datetime_offset = datetime.timedelta(days=day)
with freeze_time(first_test_date + datetime_offset, tick=True):
if day in terms_to_add:
for _ in range(terms_to_add[day]):
ff.add_term("front", "back")
# making sure the cards we are going to test are in the correct buckets
todays_cards = ff.todays_cards()
todays_tests = [test[day] for test in tests]
for card in todays_cards:
card_idx = int(card) - 1
assert todays_tests[card_idx] is not None
# test the cards
for card in range(1, 7):
test_result = tests[card - 1][day]
if test_result is not None:
ff.update_bucket(str(card), test_result)
# assert the buckets
buckets_df = ff.get_buckets()
assert buckets_df["bucket_after_test"].tolist() == [
str(card[day + 1]) for card in buckets if card[day + 1] != 0
]
def test_todays_buckets(date, expected, freezer):
# assuming a ff.start_date of 2019-12-31
with freeze_time(date, tick=True):
ff = flippyflop.FlippyFlop(dummy_service, TEST_SPREADSHEET_ID)
result = ff._todays_buckets()
assert result == expected