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