def test_load_as_matrix_tempfile(self): """ Test that whole image is loaded successfully using tempfile loader. """ wrapped_temp_loader = mock.MagicMock(wraps=load_dataset_tempfile) wrapped_vsimem_loader = mock.MagicMock(wraps=load_dataset_vsimem) with mock.patch.dict(GdalImageReader.LOAD_METHOD_CONTEXTMANAGERS, { GdalImageReader.LOAD_METHOD_TEMPFILE: wrapped_temp_loader, GdalImageReader.LOAD_METHOD_VSIMEM: wrapped_vsimem_loader }): # Using tempfile load method reader = GdalImageReader( load_method=GdalImageReader.LOAD_METHOD_TEMPFILE ) mat = reader._load_as_matrix(GH_FILE_ELEMENT) assert mat.shape == (600, 512, 3) wrapped_temp_loader.assert_called_once_with(GH_FILE_ELEMENT) wrapped_vsimem_loader.assert_not_called()
def test_load_as_matrix_vsimem(self): """ Test that whole image is loaded successfully using vsimem loader. """ if LooseVersion(osgeo.__version__).version[0] < 2: pytest.skip("Skipping VSIMEM test because GDAL version < 2") wrapped_temp_loader = mock.MagicMock(wraps=load_dataset_tempfile) wrapped_vsimem_loader = mock.MagicMock(wraps=load_dataset_vsimem) with mock.patch.dict(GdalImageReader.LOAD_METHOD_CONTEXTMANAGERS, { GdalImageReader.LOAD_METHOD_TEMPFILE: wrapped_temp_loader, GdalImageReader.LOAD_METHOD_VSIMEM: wrapped_vsimem_loader }): # Using VSIMEM load method reader = GdalImageReader( load_method=GdalImageReader.LOAD_METHOD_VSIMEM ) mat = reader._load_as_matrix(GH_FILE_ELEMENT) assert mat.shape == (600, 512, 3) wrapped_temp_loader.assert_not_called() wrapped_vsimem_loader.assert_called_once_with(GH_FILE_ELEMENT)