@@ -346,38 +346,6 @@ def SampleUV2(particle, fieldset, time): # pragma: no cover
346346 assert abs (pset .lat [0 ] - 0.5 ) < 1e-9
347347
348348
349- @pytest .mark .v4remove
350- @pytest .mark .xfail (reason = "GH1918" )
351- @pytest .mark .parametrize ("datetype" , ["float" , "datetime64" ])
352- def test_timestamps (datetype , tmpdir ):
353- data1 , dims1 = generate_fieldset_data (10 , 10 , 1 , 10 )
354- data2 , dims2 = generate_fieldset_data (10 , 10 , 1 , 4 )
355- if datetype == "float" :
356- dims1 ["time" ] = np .arange (0 , 10 , 1 ) * 86400
357- dims2 ["time" ] = np .arange (10 , 14 , 1 ) * 86400
358- else :
359- dims1 ["time" ] = np .arange ("2005-02-01" , "2005-02-11" , dtype = "datetime64[D]" )
360- dims2 ["time" ] = np .arange ("2005-02-11" , "2005-02-15" , dtype = "datetime64[D]" )
361-
362- fieldset1 = FieldSet .from_data (data1 , dims1 ) # TODO : Remove from_data
363- fieldset1 .U .data [0 , :, :] = 2.0
364- fieldset1 .write (tmpdir .join ("file1" ))
365-
366- fieldset2 = FieldSet .from_data (data2 , dims2 ) # TODO : Remove from_data
367- fieldset2 .U .data [0 , :, :] = 0.0
368- fieldset2 .write (tmpdir .join ("file2" ))
369-
370- fieldset3 = FieldSet .from_parcels (tmpdir .join ("file*" ))
371- timestamps = [dims1 ["time" ], dims2 ["time" ]]
372- fieldset4 = FieldSet .from_parcels (tmpdir .join ("file*" ), timestamps = timestamps )
373- assert np .allclose (fieldset3 .U .grid .time_full , fieldset4 .U .grid .time_full )
374-
375- for d in [0 , 8 , 10 , 12 ]:
376- fieldset3 .computeTimeChunk (d * 86400.0 , 1.0 )
377- fieldset4 .computeTimeChunk (d * 86400.0 , 1.0 )
378- assert np .allclose (fieldset3 .U .data , fieldset4 .U .data )
379-
380-
381349@pytest .mark .v4remove
382350@pytest .mark .xfail (reason = "time_periodic removed in v4" )
383351@pytest .mark .parametrize ("use_xarray" , [True , False ])
0 commit comments