def test_should_weight_two_letter_keyword(): assert_equal(_weighted(b'ab'), [1,2]) assert_equal(_weighted(b'ba'), [2,1]) assert_equal(_weighted(b'za'), [2,1]) assert_equal(_weighted(b'az'), [1,2]) assert_equal(_weighted(b'rm'), [2,1])
def test_should_weight_many_letter_keyword(): keyword = b'zebras' assert_equal(_weighted(keyword), [6,3,2,4,1,5])
def test_should_weight_one_letter_keyword(): assert_equal(_weighted(b'a'), [1]) assert_equal(_weighted(b'm'), [1]) assert_equal(_weighted(b'z'), [1])