@@ -47,7 +47,6 @@ def time_file():
4747def 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