Skip to content

Commit

Permalink
Fix VectorDataset tests
Browse files Browse the repository at this point in the history
  • Loading branch information
adamjstewart committed Feb 26, 2022
1 parent e633b85 commit 18a6b1b
Showing 1 changed file with 13 additions and 2 deletions.
15 changes: 13 additions & 2 deletions tests/datasets/test_geo.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,13 @@
import os
import pickle
from pathlib import Path
from typing import Dict
from typing import Dict, Generator

import pytest
import torch
import torch.nn as nn
from _pytest.fixtures import SubRequest
from _pytest.monkeypatch import MonkeyPatch
from rasterio.crs import CRS
from torch.utils.data import ConcatDataset

Expand Down Expand Up @@ -196,7 +197,17 @@ def test_no_data(self, tmp_path: Path) -> None:

class TestVectorDataset:
@pytest.fixture
def dataset(self) -> CanadianBuildingFootprints:
def dataset(
self, monkeypatch: Generator[MonkeyPatch, None, None]
) -> CanadianBuildingFootprints:
provinces_territories = ["Alberta"]
monkeypatch.setattr( # type: ignore[attr-defined]
CanadianBuildingFootprints, "provinces_territories", provinces_territories
)
md5s = ["25091d1f051baa30d8f2026545cfb696"]
monkeypatch.setattr( # type: ignore[attr-defined]
CanadianBuildingFootprints, "md5s", md5s
)
root = os.path.join("tests", "data", "cbf")
transforms = nn.Identity() # type: ignore[attr-defined]
return CanadianBuildingFootprints(root, res=0.1, transforms=transforms)
Expand Down

0 comments on commit 18a6b1b

Please sign in to comment.