def test_zero(self, model_national):
data = model_national._model_data_original.copy()
mask = masks.zero(data, 'csp', var='resource')
dtindex = pd.DatetimeIndex([
'2005-01-01 00:00:00', '2005-01-01 01:00:00',
'2005-01-01 02:00:00', '2005-01-01 03:00:00',
'2005-01-01 04:00:00', '2005-01-01 05:00:00',
'2005-01-01 06:00:00', '2005-01-01 16:00:00',
'2005-01-01 17:00:00', '2005-01-01 18:00:00',
'2005-01-01 19:00:00', '2005-01-01 20:00:00',
'2005-01-01 21:00:00', '2005-01-01 22:00:00',
'2005-01-01 23:00:00', '2005-01-02 00:00:00',
'2005-01-02 01:00:00', '2005-01-02 02:00:00',
'2005-01-02 03:00:00', '2005-01-02 04:00:00',
'2005-01-02 05:00:00', '2005-01-02 06:00:00',
'2005-01-02 16:00:00', '2005-01-02 17:00:00',
'2005-01-02 18:00:00', '2005-01-02 19:00:00',
'2005-01-02 20:00:00', '2005-01-02 21:00:00',
'2005-01-02 22:00:00', '2005-01-02 23:00:00',
'2005-01-03 00:00:00', '2005-01-03 01:00:00',
'2005-01-03 02:00:00', '2005-01-03 03:00:00',
'2005-01-03 04:00:00', '2005-01-03 05:00:00',
'2005-01-03 06:00:00', '2005-01-03 16:00:00',
'2005-01-03 17:00:00', '2005-01-03 18:00:00',
'2005-01-03 19:00:00', '2005-01-03 20:00:00',
'2005-01-03 21:00:00', '2005-01-03 22:00:00',
'2005-01-03 23:00:00', '2005-01-04 00:00:00',
'2005-01-04 01:00:00', '2005-01-04 02:00:00',
'2005-01-04 03:00:00', '2005-01-04 04:00:00',
'2005-01-04 05:00:00', '2005-01-04 06:00:00',
'2005-01-04 16:00:00', '2005-01-04 17:00:00',
'2005-01-04 18:00:00', '2005-01-04 19:00:00',
'2005-01-04 20:00:00', '2005-01-04 21:00:00',
'2005-01-04 22:00:00', '2005-01-04 23:00:00',
'2005-01-05 00:00:00', '2005-01-05 01:00:00',
'2005-01-05 02:00:00', '2005-01-05 03:00:00',
'2005-01-05 04:00:00', '2005-01-05 05:00:00',
'2005-01-05 06:00:00', '2005-01-05 16:00:00',
'2005-01-05 17:00:00', '2005-01-05 18:00:00',
'2005-01-05 19:00:00', '2005-01-05 20:00:00',
'2005-01-05 21:00:00', '2005-01-05 22:00:00', '2005-01-05 23:00:00'
])
assert dtindex.equals(mask[0:75])
def test_zero(self, model_national):
data = model_national._model_data_original.copy()
mask = masks.zero(data, 'csp', var='resource')
dtindex = pd.DatetimeIndex([
'2005-01-01 00:00:00', '2005-01-01 01:00:00',
'2005-01-01 02:00:00', '2005-01-01 03:00:00',
'2005-01-01 04:00:00', '2005-01-01 05:00:00',
'2005-01-01 06:00:00', '2005-01-01 16:00:00',
'2005-01-01 17:00:00', '2005-01-01 18:00:00',
'2005-01-01 19:00:00', '2005-01-01 20:00:00',
'2005-01-01 21:00:00', '2005-01-01 22:00:00',
'2005-01-01 23:00:00', '2005-01-02 00:00:00',
'2005-01-02 01:00:00', '2005-01-02 02:00:00',
'2005-01-02 03:00:00', '2005-01-02 04:00:00',
'2005-01-02 05:00:00', '2005-01-02 06:00:00',
'2005-01-02 16:00:00', '2005-01-02 17:00:00',
'2005-01-02 18:00:00', '2005-01-02 19:00:00',
'2005-01-02 20:00:00', '2005-01-02 21:00:00',
'2005-01-02 22:00:00', '2005-01-02 23:00:00',
'2005-01-03 00:00:00', '2005-01-03 01:00:00',
'2005-01-03 02:00:00', '2005-01-03 03:00:00',
'2005-01-03 04:00:00', '2005-01-03 05:00:00',
'2005-01-03 06:00:00', '2005-01-03 16:00:00',
'2005-01-03 17:00:00', '2005-01-03 18:00:00',
'2005-01-03 19:00:00', '2005-01-03 20:00:00',
'2005-01-03 21:00:00', '2005-01-03 22:00:00',
'2005-01-03 23:00:00', '2005-01-04 00:00:00',
'2005-01-04 01:00:00', '2005-01-04 02:00:00',
'2005-01-04 03:00:00', '2005-01-04 04:00:00',
'2005-01-04 05:00:00', '2005-01-04 06:00:00',
'2005-01-04 16:00:00', '2005-01-04 17:00:00',
'2005-01-04 18:00:00', '2005-01-04 19:00:00',
'2005-01-04 20:00:00', '2005-01-04 21:00:00',
'2005-01-04 22:00:00', '2005-01-04 23:00:00',
'2005-01-05 00:00:00', '2005-01-05 01:00:00',
'2005-01-05 02:00:00', '2005-01-05 03:00:00',
'2005-01-05 04:00:00', '2005-01-05 05:00:00',
'2005-01-05 06:00:00', '2005-01-05 16:00:00',
'2005-01-05 17:00:00', '2005-01-05 18:00:00',
'2005-01-05 19:00:00', '2005-01-05 20:00:00',
'2005-01-05 21:00:00', '2005-01-05 22:00:00',
'2005-01-05 23:00:00'])
assert dtindex.equals(mask[0:75])