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])
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])