def test_two_houses(self): input_string = ">" self.assertEqual(2, get_num_houses_visited(input_string))
def test_back_and_forth(self): input_string = "^v^v^v^v^v" self.assertEqual(2, get_num_houses_visited(input_string))
def test_puzzle(self): self.assertEqual(2565, get_num_houses_visited(INPUT_STRING))
def test_square(self): input_string = "^>v<" self.assertEqual(4, get_num_houses_visited(input_string))