def test_rotate_image_2():
matrix = [[5, 1, 9, 11], [2, 4, 8, 10], [13, 3, 6, 7], [15, 14, 12, 16]]
expected = [[15, 13, 2, 5], [14, 3, 4, 1], [12, 6, 8, 9], [16, 7, 10, 11]]
rotate(matrix)
assert len(matrix) == len(expected)
for i in range(len(matrix)):
assert compare_lists(expected[i], matrix[i]) is True
def test_rotate_image_4():
matrix = [[1, 2], [3, 4]]
expected = [[3, 1], [4, 2]]
rotate(matrix)
assert len(matrix) == len(expected)
for i in range(len(matrix)):
assert compare_lists(expected[i], matrix[i]) is True
def test_rotate_image_1():
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
expected = [[7, 4, 1], [8, 5, 2], [9, 6, 3]]
rotate(matrix)
assert len(matrix) == len(expected)
for i in range(len(matrix)):
assert compare_lists(expected[i], matrix[i]) is True
def test_set_matrix_zeroes_2():
matrix = [[0, 1, 2, 0], [3, 4, 5, 2], [1, 3, 1, 5]]
expected = [[0, 0, 0, 0], [0, 4, 5, 0], [0, 3, 1, 0]]
set_zeroes(matrix)
assert len(expected) == len(matrix)
for i in range(len(matrix)):
assert compare_lists(expected[i], matrix[i]) is True
def test_set_matrix_zeroes_1():
matrix = [[1, 1, 1], [1, 0, 1], [1, 1, 1]]
expected = [[1, 0, 1], [0, 0, 0], [1, 0, 1]]
set_zeroes(matrix)
assert len(expected) == len(matrix)
for i in range(len(matrix)):
assert compare_lists(expected[i], matrix[i]) is True
def test_word_break_2_2():
s = "pineapplepenapple"
word_dict = ["apple", "pen", "applepen", "pine", "pineapple"]
expected = [
"pine apple pen apple", "pineapple pen apple", "pine applepen apple"
]
actual = word_break(s, word_dict)
assert compare_lists(expected, actual) is True
def test_average_of_levels_in_binary_tree_1():
root = TreeNode(3)
root.left = TreeNode(9)
root.right = TreeNode(20)
root.right.left = TreeNode(15)
root.right.right = TreeNode(7)
expected = [3.0, 14.5, 11.0]
actual = average_of_levels(root)
assert compare_lists(expected, actual) is True
def test_word_search_2_1():
board = [
["o", "a", "a", "n"],
["e", "t", "a", "e"],
["i", "h", "k", "r"],
["i", "f", "l", "v"],
]
words = ["oath", "pea", "eat", "rain"]
expected = ["eat", "oath"]
actual = find_words(board, words)
assert compare_lists(expected, actual) is True
def test_spiral_matrix_7():
matrix = [
[1, 2, 3, 4, 5],
[6, 7, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20],
]
expected = [
1, 2, 3, 4, 5, 10, 15, 20, 19, 18, 17, 16, 11, 6, 7, 8, 9, 14, 13, 12
]
actual = spiral_order(matrix)
assert compare_lists(expected, actual) is True
def test_word_search_2_4():
board = [["a"]]
words = [
"earth",
"jupiter",
"mars",
"mercury",
"neptune",
"saturn",
"uranus",
"venus",
"a",
]
expected = ["a"]
actual = find_words(board, words)
assert compare_lists(expected, actual) is True
def test_word_break_2_10():
s = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa "
word_dict = [
"a",
"aa",
"aaa",
"aaaa",
"aaaaa",
"aaaaaa",
"aaaaaaa",
"aaaaaaaa",
"aaaaaaaaa",
"aaaaaaaaaa",
]
expected = []
actual = word_break(s, word_dict)
assert compare_lists(expected, actual) is True
def test_word_break_2_9():
s = "ifindleetcodefun"
word_dict = [
"i",
"leet",
"leetcode",
"code",
"fun",
"find",
"codefun",
"test",
"do",
]
expected = [
"i find leetcode fun", "i find leet code fun", "i find leet codefun"
]
actual = word_break(s, word_dict)
assert compare_lists(expected, actual) is True
def test_word_search_2_3():
board = [
["j", "s", "o", "l", "u", "t", "i", "s"],
["s", "u", "n", "a", "r", "u", "u", "a"],
["n", "e", "p", "t", "u", "n", "e", "t"],
["s", "o", "n", "i", "e", "i", "s", "u"],
["r", "c", "e", "v", "t", "r", "e", "r"],
["a", "h", "t", "r", "a", "e", "s", "n"],
["m", "m", "e", "r", "c", "u", "r", "y"],
]
words = [
"earth",
"jupiter",
"mars",
"mercury",
"neptune",
"saturn",
"uranus",
"venus",
]
expected = ["earth", "mars", "mercury", "neptune", "saturn", "uranus"]
actual = find_words(board, words)
assert compare_lists(expected, actual) is True
def test_index_pairs_of_a_string_1():
text = "thestoryofleetcodeandme"
words = ["story", "fleet", "leetcode"]
expected = [[3, 7], [9, 13], [10, 17]]
actual = index_pairs(text, words)
assert compare_lists(expected, actual, DEBUG) is True
def test_find_all_duplicates_in_an_array_2():
nums = [1, 1]
expected = [1]
actual = find_duplicates(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_spiral_matrix_2():
matrix = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]
expected = [1, 2, 3, 4, 8, 12, 11, 10, 9, 5, 6, 7]
actual = spiral_order(matrix)
assert compare_lists(expected, actual) is True
def test_product_of_array_except_self_5():
nums = [1, 0]
expected = [0, 1]
actual = product_except_self(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_squares_of_a_sorted_array_1():
nums = [-4, -1, 0, 3, 10]
expected = [0, 1, 9, 16, 100]
actual = sorted_squares(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_word_search_2_2():
board = [["a", "b"], ["c", "d"]]
words = ["abcb"]
expected = []
actual = find_words(board, words)
assert compare_lists(expected, actual) is True
def test_spiral_matrix_1():
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
expected = [1, 2, 3, 6, 9, 8, 7, 4, 5]
actual = spiral_order(matrix)
assert compare_lists(expected, actual) is True
def test_spiral_matrix_6():
matrix = [[1], [2]]
expected = [1, 2]
actual = spiral_order(matrix)
assert compare_lists(expected, actual) is True
def test_squares_of_a_sorted_array_2():
nums = [-7, -3, 2, 3, 11]
expected = [4, 9, 9, 49, 121]
actual = sorted_squares(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_squares_of_a_sorted_array_5():
nums = [0]
expected = [0]
actual = sorted_squares(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_find_all_duplicates_in_an_array_4():
nums = [1, 4, 5, 6, 9, 3, 4, 5, 1, 6]
expected = [4, 5, 1, 6]
actual = find_duplicates(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_product_of_array_except_self_2():
nums = [-1, 1, 0, -3, 3]
expected = [0, 0, 9, 0, 0]
actual = product_except_self(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_course_schedule_2_2():
num_courses = 4
pre_reqs = [[1, 0], [2, 0], [3, 1], [3, 2]]
expected = [0, 2, 1, 3]
actual = find_order(num_courses, pre_reqs)
assert compare_lists(actual, expected) is True
def test_product_of_array_except_self_1():
nums = [1, 2, 3, 4]
expected = [24, 12, 8, 6]
actual = product_except_self(nums)
assert compare_lists(expected, actual, DEBUG) is True
def test_course_schedule_2_3():
num_courses = 1
pre_reqs = []
expected = [0]
actual = find_order(num_courses, pre_reqs)
assert compare_lists(actual, expected) is True
def test_course_schedule_2_1():
num_courses = 2
pre_reqs = [[1, 0]]
expected = [0, 1]
actual = find_order(num_courses, pre_reqs)
assert compare_lists(actual, expected) is True
def test_find_all_duplicates_in_an_array_1():
nums = [4, 3, 2, 7, 8, 2, 3, 1]
expected = [2, 3]
actual = find_duplicates(nums)
assert compare_lists(expected, actual, DEBUG) is True
