def test_reads_values_from_input_dataset_with_projection_when_no_options(self, mock_osr):
expected_srs = mock.MagicMock()
expected_wkt = "expected_wkt"
self.mock_dataset.GetProjection = mock.MagicMock(return_value=expected_wkt)
mock_osr.SpatialReference = mock.MagicMock(return_value=expected_srs)
input_srs, input_srs_wkt = gdal2tiles.setup_input_srs(
self.mock_dataset, self.DEFAULT_ATTRDICT_OPTIONS)
self.assertEqual(input_srs, expected_srs)
self.assertEqual(input_srs_wkt, expected_wkt)
mock_osr.SpatialReference().ImportFromWkt.assert_called_with(expected_wkt)
def test_uses_the_passed_arguments_in_priority(self, mock_osr):
option_srs = "o_srs"
self.DEFAULT_ATTRDICT_OPTIONS['s_srs'] = option_srs
expected_srs = mock.MagicMock()
expected_wkt = "expected_wkt"
mock_osr.SpatialReference = mock.MagicMock(return_value=expected_srs)
mock_osr.SpatialReference().ExportToWkt = mock.MagicMock(return_value=expected_wkt)
input_srs, input_srs_wkt = gdal2tiles.setup_input_srs(
self.mock_dataset, self.DEFAULT_ATTRDICT_OPTIONS)
self.assertEqual(input_srs, expected_srs)
self.assertEqual(input_srs_wkt, expected_wkt)
mock_osr.SpatialReference().SetFromUserInput.assert_called_with(option_srs)
def test_reads_values_from_input_dataset_with_projection_when_no_options(
self, mock_osr):
expected_srs = mock.MagicMock()
expected_wkt = "expected_wkt"
self.mock_dataset.GetProjection = mock.MagicMock(
return_value=expected_wkt)
mock_osr.SpatialReference = mock.MagicMock(return_value=expected_srs)
input_srs, input_srs_wkt = gdal2tiles.setup_input_srs(
self.mock_dataset, self.DEFAULT_ATTRDICT_OPTIONS)
self.assertEqual(input_srs, expected_srs)
self.assertEqual(input_srs_wkt, expected_wkt)
mock_osr.SpatialReference().ImportFromWkt.assert_called_with(
expected_wkt)
def test_uses_the_passed_arguments_in_priority(self, mock_osr):
option_srs = "o_srs"
self.DEFAULT_ATTRDICT_OPTIONS['s_srs'] = option_srs
expected_srs = mock.MagicMock()
expected_wkt = "expected_wkt"
mock_osr.SpatialReference = mock.MagicMock(return_value=expected_srs)
mock_osr.SpatialReference().ExportToWkt = mock.MagicMock(
return_value=expected_wkt)
input_srs, input_srs_wkt = gdal2tiles.setup_input_srs(
self.mock_dataset, self.DEFAULT_ATTRDICT_OPTIONS)
self.assertEqual(input_srs, expected_srs)
self.assertEqual(input_srs_wkt, expected_wkt)
mock_osr.SpatialReference().SetFromUserInput.assert_called_with(
option_srs)