def test_combine_f107_all(self): """Test combine_f107 when all input is provided with '' and '45day'""" f107_inst = mm_f107.combine_f107(self.combineInst[''], self.combineInst['45day'], **self.combineTimes) assert f107_inst.index[0] >= self.combineTimes['start'] assert f107_inst.index[-1] < self.combineTimes['stop'] assert len(f107_inst.data.columns) == 1 assert f107_inst.data.columns[0] == 'f107' del f107_inst
def test_combine_f107_no_data(self): """Test combine_f107 when no data is present for specified times""" combo_in = { kk: self.combine_inst['forecast'] for kk in ['standard_inst', 'forecast_inst'] } combo_in['start'] = dt.datetime(2014, 2, 19) combo_in['stop'] = dt.datetime(2014, 2, 24) f107_inst = mm_f107.combine_f107(**combo_in) assert f107_inst.data.isnull().all()["f107"] del combo_in, f107_inst
def test_combine_f107_inst_time(self): """Test combine_f107 with times provided through datasets""" self.combineInst['all'].load(date=self.combineTimes['start']) self.combineInst['forecast'].load(date=self.test_day) f107_inst = mm_f107.combine_f107(self.combineInst['all'], self.combineInst['forecast']) assert f107_inst.index[0] == dt.datetime(1947, 2, 13) assert f107_inst.index[-1] <= self.combineTimes['stop'] assert len(f107_inst.data.columns) == 1 assert f107_inst.data.columns[0] == 'f107' del f107_inst
def test_combine_f107_inst_time(self): """Test combine_f107 with times provided through datasets""" self.combine_inst['historic'].load( date=self.combine_inst['historic'].lasp_stime, end_date=self.combine_times['start']) self.combine_inst['forecast'].load(date=self.test_day) f107_inst = mm_f107.combine_f107(self.combine_inst['historic'], self.combine_inst['forecast']) assert f107_inst.index[0] == self.combine_inst['historic'].lasp_stime assert f107_inst.index[-1] <= self.combine_times['stop'] assert len(f107_inst.data.columns) == 1 assert f107_inst.data.columns[0] == 'f107' del f107_inst
def test_combine_f107_no_time(self): """Test combine_f107 failure when no times are provided""" with pytest.raises(ValueError): mm_f107.combine_f107(self.combine_inst['historic'], self.combine_inst['forecast'])
def test_combine_f107_none(self): """ Test combine_f107 failure when no input is provided""" with pytest.raises(TypeError): mm_f107.combine_f107()