def test_drop(self, model_national):
data = model_national._model_data_original.copy()
timesteps = ['2005-01-01 23:00', '2005-01-01 22:00']
data_dropped = funcs.drop(data, timesteps)
assert len(data_dropped.timesteps) == 742
result_timesteps = list(data_dropped.coords['timesteps'].values)
assert '2005-01-01 21:00' not in result_timesteps
assert '2005-01-01 22:00' not in result_timesteps
def test_drop(self, model_national):
data = model_national._model_data_original.copy()
timesteps = ['2005-01-01 23:00', '2005-01-01 22:00']
data_dropped = funcs.drop(data, timesteps)
assert len(data_dropped.timesteps) == 742
result_timesteps = list(data_dropped.coords['timesteps'].values)
assert '2005-01-01 21:00' not in result_timesteps
assert '2005-01-01 22:00' not in result_timesteps
def test_drop_invalid_timesteps(self, model_national):
data = model_national._model_data_original.copy()
timesteps = ['XXX2005-01-01 23:00']
with pytest.raises(exceptions.ModelError):
funcs.drop(data, timesteps)
def test_drop_invalid_timesteps(self, model_national):
data = model_national._model_data_original.copy()
timesteps = ['XXX2005-01-01 23:00']
with pytest.raises(exceptions.ModelError):
funcs.drop(data, timesteps)
