def test_word_squares(self): self.assertEqual([['wall', 'area', 'lead', 'lady'], ['ball', 'area', 'lead', 'lady']], \ word_squares(["area", "lead", "wall", "lady", "ball"]))
# l a d y # Note: # There are at least 1 and at most 1000 words. # All words will have the exact same length. # Word length is at least 1 and at most 5. # Each word contains only lowercase English alphabet a-z. # Example 1: # Input: # ["area","lead","wall","lady","ball"] # Output: # [ # [ "wall", # "area", # "lead", # "lady" # ], # [ "ball", # "area", # "lead", # "lady" # ] # ] from algorithms.strings import word_squares a = ["area", "lead", "wall", "lady", "ball"] print(word_squares(a))
def test_word_squares(self): self.assertEqual( [["wall", "area", "lead", "lady"], ["ball", "area", "lead", "lady"]], word_squares(["area", "lead", "wall", "lady", "ball"]), )