This commit is contained in:
Adam J. Stewart 2021-07-21 01:47:39 +00:00
Родитель 3b8af13fec
Коммит 4edfa1036e
2 изменённых файлов: 11 добавлений и 1 удалений

Просмотреть файл

@ -3,6 +3,7 @@ import shutil
from pathlib import Path
from typing import Generator
import matplotlib.pyplot as plt
import pytest
import torch
import torchvision.datasets.utils
@ -36,6 +37,9 @@ class TestCDL:
monkeypatch.setattr(CDL, "md5s", md5s) # type: ignore[attr-defined]
url = os.path.join("tests", "data", "cdl", "{}_30m_cdls.zip")
monkeypatch.setattr(CDL, "url", url) # type: ignore[attr-defined]
monkeypatch.setattr( # type: ignore[attr-defined]
plt, "show", lambda *args: None
)
(tmp_path / "cdl").mkdir()
root = str(tmp_path)
transforms = Identity()

Просмотреть файл

@ -1,8 +1,11 @@
import os
from pathlib import Path
from typing import Generator
import matplotlib.pyplot as plt
import pytest
import torch
from pytest import MonkeyPatch
from rasterio.crs import CRS
from torchgeo.datasets import BoundingBox, Landsat8, ZipDataset
@ -11,7 +14,10 @@ from torchgeo.transforms import Identity
class TestLandsat8:
@pytest.fixture
def dataset(self) -> Landsat8:
def dataset(self, monkeypatch: Generator[MonkeyPatch, None, None]) -> Landsat8:
monkeypatch.setattr( # type: ignore[attr-defined]
plt, "show", lambda *args: None
)
root = os.path.join("tests", "data")
bands = ["B4", "B3", "B2"]
transforms = Identity()