Skip to content

Commit

Permalink
reformat with black
Browse files Browse the repository at this point in the history
  • Loading branch information
tgoelles committed Sep 28, 2023
1 parent 3304207 commit 9201b31
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions kedro-datasets/tests/xarray/test_geotiff_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,20 +10,22 @@
def cog_file_path():
return Path(__file__).parent / "cog.tif"


@pytest.fixture
def geotiff_dataset(cog_file_path, save_args, fs_args) -> GeoTiffDataset:
return GeoTiffDataset(filepath=cog_file_path, save_args=save_args, fs_args=fs_args)


@pytest.fixture
def versioned_geotiff_dataset(filepath_yaml, load_version, save_version) -> GeoTiffDataset:
def versioned_geotiff_dataset(
filepath_yaml, load_version, save_version
) -> GeoTiffDataset:
return GeoTiffDataset(
filepath=filepath_yaml, version=Version(load_version, save_version)
)



def test_save_and_load(self, geotiff_dataset, cog_file_path):
"""Test saving and reloading the data set."""
loaded_tiff = geotiff_dataset.load()
assert type(loaded_tiff) == rioxarray.raster_dataset.RasterArray

0 comments on commit 9201b31

Please sign in to comment.