def test_get_consecutive_words_4_word(): """get_consecutive_words should return correct words for 4-word blocks""" nose.assert_list_equal( sim.get_consecutive_words(21, num_words_per_block=4), [20, 21, 22, 23])
def test_get_consecutive_words_1_word(): """get_consecutive_words should return same word for 1-word blocks""" nose.assert_list_equal( sim.get_consecutive_words(23, num_words_per_block=1), [23])
def test_get_consecutive_words_2_word(): """get_consecutive_words should return correct words for 2-word blocks""" nose.assert_list_equal( sim.get_consecutive_words(22, num_words_per_block=2), [22, 23])