def test_ensure_valid_time(da1, da3): res1 = cfcoords.ensure_valid_time(da1.squeeze()) res2 = cfcoords.ensure_valid_time(res1) assert 'valid_time' in res1.coords assert res2 is res1 res1 = cfcoords.ensure_valid_time(da3.isel(ref_time=0).squeeze()) res2 = cfcoords.ensure_valid_time(res1) assert 'valid_time' in res1.coords assert res2 is res1 res1 = cfcoords.ensure_valid_time(da3.squeeze()) assert 'valid_time' in res1.coords with pytest.raises(ValueError): cfcoords.ensure_valid_time(da3.mean(dim='ref_time').squeeze())
def test_ensure_valid_time(da1, da3): res1 = cfcoords.ensure_valid_time(da1.squeeze()) res2 = cfcoords.ensure_valid_time(res1) assert 'valid_time' in res1.coords assert res2 is res1 res1 = cfcoords.ensure_valid_time(da3.isel(ref_time=0).squeeze()) res2 = cfcoords.ensure_valid_time(res1) assert 'valid_time' in res1.coords assert res2 is res1 res1 = cfcoords.ensure_valid_time(da3.squeeze()) assert 'valid_time' in res1.coords coords1 = res1.coords assert coords1['valid_time'].shape == (coords1['ref_time'].size, coords1['step'].size) with pytest.raises(ValueError): cfcoords.ensure_valid_time(da3.mean(dim='ref_time').squeeze())