def test_Plotter_plot(source, image_mock): ex = ons.Extent(1100, 1900, 4200, 5500) plotter = ons.Plotter(ex, source) ax = mock.Mock() plotter.plot(ax, bob="fish") assert source.call_args_list == [ mock.call("SV 1000 4000"), mock.call("SV 1000 5000") ] assert ax.imshow.call_args_list == [ mock.call(image_mock.new.return_value, interpolation="lanczos", extent=(1000, 2000, 4000, 6000), bob="fish"), ]
def test_Plotter_ignore_errors(source, image_mock): source.side_effect = Exception ex = ons.Extent(1100, 1900, 4200, 5500) plotter = ons.Plotter(ex, source) ax = mock.Mock() plotter.plotlq(ax, bob="fish") assert ax.imshow.call_args_list == [ mock.call(source.blank.return_value, interpolation="lanczos", extent=(1000, 2000, 4000, 5000), bob="fish"), mock.call(source.blank.return_value, interpolation="lanczos", extent=(1000, 2000, 5000, 6000), bob="fish") ]
def test_Plotter_as_one_image(source, image_mock): ex = ons.Extent(1100, 1900, 4200, 5500) plotter = ons.Plotter(ex, source) ax = mock.Mock() out = plotter.as_one_image() assert source.call_args_list == [ mock.call("SV 1000 4000"), mock.call("SV 1000 5000") ] image_mock.new.assert_called_with("RGB", (2000, 4000)) im = image_mock.new.return_value assert out is im assert im.paste.call_args_list == [ mock.call(source.return_value, (0, 2000)), mock.call(source.return_value, (0, 0)) ]