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)
