Function collection_with_sessions

backend/tests/stats/test_api.py:15–25

(db)

@pytest.fixture
def collection_with_sessions(db):
    def _create(user_id, num_cards=5, num_sessions=10):
        collection = Collection(name="Test Collection", user_id=user_id)
        db.add(collection)
        db.flush()
        cards = create_cards(db, collection, num_cards)
        sessions = create_sessions(db, user_id, collection, num_sessions, num_cards)
        create_practice_cards(db, sessions, cards)
        return collection

    return _create


@pytest.mark.parametrize("limit", [1, 10, 30])