Skip to content

Commit 8ffc894

Browse files
committed
Clean up tests
1 parent 88b7db7 commit 8ffc894

File tree

2 files changed

+13
-42
lines changed

2 files changed

+13
-42
lines changed

tests/test_dfs1.py

Lines changed: 9 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -5,17 +5,14 @@
55

66
import mikeio
77

8-
from mikeio import EUMType, EUMUnit
9-
108

119
def test_filenotexist() -> None:
1210
with pytest.raises(FileNotFoundError):
1311
mikeio.Dfs1("file_that_does_not_exist.dfs1")
1412

1513

1614
def test_repr() -> None:
17-
filename = r"tests/testdata/random.dfs1"
18-
dfs = mikeio.Dfs1(filename)
15+
dfs = mikeio.Dfs1("tests/testdata/random.dfs1")
1916

2017
text = repr(dfs)
2118

@@ -25,8 +22,7 @@ def test_repr() -> None:
2522

2623

2724
def test_properties() -> None:
28-
filename = r"tests/testdata/tide1.dfs1"
29-
dfs = mikeio.Dfs1(filename)
25+
dfs = mikeio.Dfs1("tests/testdata/tide1.dfs1")
3026

3127
assert dfs.dx == 0.06666692346334457
3228
assert dfs.x0 == 0.0
@@ -47,9 +43,7 @@ def test_properties() -> None:
4743

4844

4945
def test_read_write_properties(tmp_path: Path) -> None:
50-
# test that properties are the same after read-write
51-
filename = r"tests/testdata/tide1.dfs1"
52-
ds1 = mikeio.read(filename)
46+
ds1 = mikeio.read("tests/testdata/tide1.dfs1")
5347

5448
fp = tmp_path / "tide1.dfs1"
5549
ds1.to_dfs(fp)
@@ -59,26 +53,23 @@ def test_read_write_properties(tmp_path: Path) -> None:
5953

6054

6155
def test_read() -> None:
62-
filename = r"tests/testdata/random.dfs1"
63-
dfs = mikeio.Dfs1(filename)
56+
dfs = mikeio.Dfs1("tests/testdata/random.dfs1")
6457

6558
ds = dfs.read(items=[0])
6659
data = ds[0].to_numpy()
6760
assert data.shape == (100, 3) # time, x
6861

6962

7063
def test_read_item_names() -> None:
71-
filename = r"tests/testdata/random.dfs1"
72-
dfs = mikeio.Dfs1(filename)
64+
dfs = mikeio.Dfs1("tests/testdata/random.dfs1")
7365

7466
ds = dfs.read(items=["testing water level"])
7567
data = ds[0].to_numpy()
7668
assert data.shape == (100, 3) # time, x
7769

7870

7971
def test_read_time_steps() -> None:
80-
filename = r"tests/testdata/random.dfs1"
81-
dfs = mikeio.Dfs1(filename)
72+
dfs = mikeio.Dfs1("tests/testdata/random.dfs1")
8273

8374
ds = dfs.read(time=[3, 5])
8475
data = ds[0].to_numpy()
@@ -102,27 +93,12 @@ def test_write_some_time_steps_new_file(tmp_path: Path) -> None:
10293

10394

10495
def test_read_item_names_not_in_dataset_fails() -> None:
105-
filename = r"tests/testdata/random.dfs1"
106-
dfs = mikeio.Dfs1(filename)
96+
dfs = mikeio.Dfs1("tests/testdata/random.dfs1")
10797

10898
with pytest.raises(KeyError):
10999
dfs.read(items=["NOTAREALVARIABLE"])
110100

111101

112-
def test_read_names_access() -> None:
113-
filename = r"tests/testdata/random.dfs1"
114-
dfs = mikeio.Dfs1(filename)
115-
116-
res = dfs.read(items=[0])
117-
item_data = res[0].to_numpy()
118-
time = res.time
119-
assert item_data.shape == (100, 3) # time, x
120-
assert len(time) == 100
121-
assert res.items[0].name == "testing water level"
122-
assert res.items[0].type == EUMType.Water_Level
123-
assert res.items[0].unit == EUMUnit.meter
124-
125-
126102
def test_read_start_end_time() -> None:
127103
dfs = mikeio.Dfs1("tests/testdata/random.dfs1")
128104
ds = dfs.read()
@@ -140,15 +116,13 @@ def test_read_start_end_time_relative_time() -> None:
140116

141117

142118
def test_get_time_axis_without_reading_data() -> None:
143-
dfs0file = r"tests/testdata/random.dfs1"
144-
dfs = mikeio.Dfs1(dfs0file)
119+
dfs = mikeio.Dfs1("tests/testdata/random.dfs1")
145120
assert isinstance(dfs.time, pd.DatetimeIndex)
146121
assert len(dfs.time) == 100
147122

148123

149124
def test_get_time_axis_without_reading_data_relative() -> None:
150-
dfs0file = r"tests/testdata/physical_basin_wave_maker_signal.dfs1"
151-
dfs = mikeio.Dfs1(dfs0file)
125+
dfs = mikeio.Dfs1("tests/testdata/physical_basin_wave_maker_signal.dfs1")
152126
assert isinstance(dfs.time, pd.DatetimeIndex) # start time is not correct !
153127
assert len(dfs.time) == 200
154128

tests/test_dfs2.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -185,18 +185,16 @@ def test_read_bad_item(dfs2_random: Dfs2) -> None:
185185

186186

187187
def test_read_temporal_subset_slice() -> None:
188-
filename = r"tests/testdata/eq.dfs2"
189-
dfs = mikeio.Dfs2(filename)
188+
dfs = mikeio.Dfs2("tests/testdata/eq.dfs2")
190189
ds = dfs.read(time=slice("2000-01-01 00:00", "2000-01-01 12:00"))
191190

192191
assert len(ds.time) == 13
193192

194193

195194
def test_read_area_subset_bad_bbox() -> None:
196-
filename = "tests/testdata/europe_wind_long_lat.dfs2"
197-
bbox = (10, 40, 20)
195+
bbox = (10, 40, 20) # bbox should be 4 elements long
198196
with pytest.raises(ValueError):
199-
mikeio.read(filename, area=bbox)
197+
mikeio.read("tests/testdata/europe_wind_long_lat.dfs2", area=bbox)
200198

201199

202200
def test_read_area_subset_geo() -> None:
@@ -216,8 +214,7 @@ def test_read_area_subset_geo() -> None:
216214

217215

218216
def test_subset_bbox() -> None:
219-
filename = "tests/testdata/europe_wind_long_lat.dfs2"
220-
ds = mikeio.read(filename)
217+
ds = mikeio.read("tests/testdata/europe_wind_long_lat.dfs2")
221218
dssel = ds.sel(area=ds.geometry.bbox) # this is the entire area
222219
assert ds.geometry == dssel.geometry
223220

0 commit comments

Comments
 (0)