def test_green_between(mock_geolocate,mock_location,mock_count): mock_location.return_value = [[1,0],[2,0]] with patch.object(Map,'__init__') as mock_get: trail_graph = Greengraph(start,end) mock_get.return_value = None trail_graph.green_between(1) print mock_get.mock_calls mock_get.assert_has_calls([call(1, 0), call(2, 0)])
def test_green_between(mock_count_green, mock_geolocate): # Test Map objects created during green_between method are called with the # expected parameters with patch.object(Greengraph, "location_sequence") as mock_location_sequence: mock_location_sequence.return_value = [ [0., 0.], [1., 1.], [2., 2.], [3., 3.]] expected_calls = [call(0.0, 0.0), call( 1.0, 1.0), call(2.0, 2.0), call(3.0, 3.0)] with patch.object(Map, "__init__") as mock_Map_init: mock_Map_init.return_value = None test_Greengraph = Greengraph(start, end) test_Greengraph.green_between(6) mock_Map_init.assert_has_calls(expected_calls)
def process(): parser = ArgumentParser( description="generates a graph of the proportion of green pixels in a series of satellite images between two points." ) parser.add_argument("--from", dest="start", default="London") parser.add_argument("--to", default="Oxford") parser.add_argument("--steps", default=20) parser.add_argument("--out", default="greengraph.jpg") arguments = parser.parse_args() mygraph = Greengraph(arguments.start, arguments.to) data = mygraph.green_between(arguments.steps) plt.plot(data) savefig(arguments.out)