Skip to content

Commit 3246316

Browse files
committed
Remove memory profiling file from the tests
1 parent 735a0d1 commit 3246316

File tree

1 file changed

+4
-17
lines changed

1 file changed

+4
-17
lines changed

test/test_xarray_io.py

Lines changed: 4 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@ def time_file():
4747
def profile_function(
4848
func: Callable,
4949
args: Tuple[Any, ...],
50-
memory_file,
5150
time_file,
5251
print_args={},
5352
) -> None:
@@ -81,15 +80,12 @@ def create_and_write_jinja(tmp_path, data: xr.DataArray):
8180
test_combinations,
8281
)
8382
@pytest.mark.skip("Too slow for large data")
84-
def test_xarray_to_text_jinja(
85-
tmp_path, max_size, chunks, time_file, memory_file
86-
):
83+
def test_xarray_to_text_jinja(tmp_path, max_size, chunks, time_file):
8784
data = xr.DataArray(da.arange(0, max_size, 1), dims="x")
8885
data = data.chunk(chunks)
8986
profile_function(
9087
create_and_write_jinja,
9188
(tmp_path, data),
92-
memory_file,
9389
time_file,
9490
print_args={"max_size": max_size, "chunks": chunks},
9591
)
@@ -119,15 +115,12 @@ def create_and_write_pandas(tmp_path, data: xr.DataArray):
119115
"max_size,chunks",
120116
test_combinations,
121117
)
122-
def test_xarray_to_text_pandas(
123-
tmp_path, max_size, chunks, memory_file, time_file
124-
):
118+
def test_xarray_to_text_pandas(tmp_path, max_size, chunks, time_file):
125119
data = xr.DataArray(da.arange(0, max_size, 1), dims="x")
126120
data = data.chunk(chunks)
127121
profile_function(
128122
create_and_write_pandas,
129123
(tmp_path, data),
130-
memory_file,
131124
time_file,
132125
print_args={"max_size": max_size, "chunks": chunks},
133126
)
@@ -151,15 +144,12 @@ def create_and_write_np_savetxt(tmp_path, data: xr.DataArray):
151144
test_combinations,
152145
)
153146
@pytest.mark.skip("Too slow for large data")
154-
def test_xarray_to_text_np_savetxt(
155-
tmp_path, max_size, chunks, memory_file, time_file
156-
):
147+
def test_xarray_to_text_np_savetxt(tmp_path, max_size, chunks, time_file):
157148
data = xr.DataArray(da.arange(0, max_size, 1), dims="x")
158149
data = data.chunk(chunks)
159150
profile_function(
160151
create_and_write_np_savetxt,
161152
(tmp_path, data),
162-
memory_file,
163153
time_file,
164154
print_args={"max_size": max_size, "chunks": chunks},
165155
)
@@ -194,15 +184,12 @@ def create_and_write_extras(tmp_path, data: xr.DataArray):
194184
"max_size,chunks",
195185
test_combinations,
196186
)
197-
def test_xarray_to_text_extras(
198-
tmp_path, max_size, chunks, memory_file, time_file
199-
):
187+
def test_xarray_to_text_extras(tmp_path, max_size, chunks, time_file):
200188
data = xr.DataArray(da.arange(0, max_size, 1), dims="x")
201189
data = data.chunk(chunks)
202190
profile_function(
203191
create_and_write_extras,
204192
(tmp_path, data),
205-
memory_file,
206193
time_file,
207194
print_args={"max_size": max_size, "chunks": chunks},
208195
)

0 commit comments

Comments
 (0)