зеркало из https://github.com/microsoft/torchgeo.git
Don't actually display plots
This commit is contained in:
Родитель
3b8af13fec
Коммит
4edfa1036e
|
@ -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()
|
||||
|
|
Загрузка…
Ссылка в новой задаче