(source, target, use_cftime, freq)
| 23 | ], |
| 24 | ) |
| 25 | def test_convert_calendar(source, target, use_cftime, freq): |
| 26 | src = DataArray( |
| 27 | date_range("2004-01-01", "2004-12-31", freq=freq, calendar=source), |
| 28 | dims=("time",), |
| 29 | name="time", |
| 30 | ) |
| 31 | da_src = DataArray( |
| 32 | np.linspace(0, 1, src.size), dims=("time",), coords={"time": src} |
| 33 | ) |
| 34 | |
| 35 | conv = convert_calendar(da_src, target, use_cftime=use_cftime) |
| 36 | |
| 37 | assert conv.time.dt.calendar == target |
| 38 | |
| 39 | if source != "noleap": |
| 40 | expected_times = date_range( |
| 41 | "2004-01-01", |
| 42 | "2004-12-31", |
| 43 | freq=freq, |
| 44 | use_cftime=use_cftime, |
| 45 | calendar=target, |
| 46 | ) |
| 47 | else: |
| 48 | expected_times_pre_leap = date_range( |
| 49 | "2004-01-01", |
| 50 | "2004-02-28", |
| 51 | freq=freq, |
| 52 | use_cftime=use_cftime, |
| 53 | calendar=target, |
| 54 | ) |
| 55 | expected_times_post_leap = date_range( |
| 56 | "2004-03-01", |
| 57 | "2004-12-31", |
| 58 | freq=freq, |
| 59 | use_cftime=use_cftime, |
| 60 | calendar=target, |
| 61 | ) |
| 62 | expected_times = expected_times_pre_leap.append(expected_times_post_leap) |
| 63 | np.testing.assert_array_equal(conv.time, expected_times) |
| 64 | |
| 65 | |
| 66 | def test_convert_calendar_dataset(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…