def test_get_translation_variants(self):
a, b, c, d = insert_words(*'abcd')
q.record_answer(a, b)
res = q.get_translation_variants(a, 3)
self.assertIn(b, res)
self.assertIn(a, res)
self.assertEqual(len(res), 3)
def cram(count):
words_ids = q.select_most_hard_words_ids(count)
for word_id in words_ids:
variant_ids = q.get_translation_variants(word_id, 4)
word, t = q.get_word_and_translation(word_id)
variants = [q.get_word_and_translation(id)[1] for id in variant_ids]
answer = choice_question(word, variants)
answer_id = variant_ids[answer]
q.record_answer(word_id, variant_ids[answer])
if answer_id == word_id:
print(_("Correct!"))
else:
print(_('Wrong! "{word}" is correctly translated as "{translation}"').format(word=word, translation=t))
