def test_parse_obssumm_dbase_file(): """ Ensure that all required data are extracted from the RHESSI observing summary database file mocked in `hessi_data()` """ mock_file = mock.mock_open() mock_file.return_value.__iter__.return_value = hessi_data() dbase_data = {} with mock.patch('sunpy.instr.rhessi.open', mock_file, create=True): dbase_data = rhessi.parse_obssumm_dbase_file(None) assert len(dbase_data.keys()) == 7 # verify each of the 7 fields assert dbase_data['filename'] == [ 'hsi_obssumm_19721101_139.fit', 'hsi_obssumm_19721102_144.fit' ] assert dbase_data['orb_st'] == [7, 9] assert dbase_data['orb_end'] == [8, 10] assert dbase_data['start_time'] == [ datetime(1972, 11, 1, 0, 0), datetime(1972, 11, 2, 0, 0) ] assert dbase_data['end_time'] == [ datetime(1972, 11, 2, 0, 0), datetime(1972, 11, 3, 0, 0) ] assert dbase_data['status_flag'] == [3, 4] assert dbase_data['npackets'] == [2, 1]
def test_parse_obssumm_dbase_file(): """ Ensure that all required data are extracted from the RHESSI observing summary database file mocked in `hessi_data()` """ mock_file = mock.mock_open() mock_file.return_value.__iter__.return_value = hessi_data() dbase_data = {} with mock.patch('sunpy.instr.rhessi.open', mock_file, create=True): dbase_data = rhessi.parse_obssumm_dbase_file(None) assert len(dbase_data.keys()) == 7 # verify each of the 7 fields assert dbase_data['filename'] == ['hsi_obssumm_19721101_139.fit', 'hsi_obssumm_19721102_144.fit'] assert dbase_data['orb_st'] == [7, 9] assert dbase_data['orb_end'] == [8, 10] assert dbase_data['start_time'] == [datetime(1972, 11, 1, 0, 0), datetime(1972, 11, 2, 0, 0)] assert dbase_data['end_time'] == [datetime(1972, 11, 2, 0, 0), datetime(1972, 11, 3, 0, 0)] assert dbase_data['status_flag'] == [3, 4] assert dbase_data['npackets'] == [2, 1]
def test_parse_obssum_dbase_file(): file = rhessi.get_obssumm_dbase_file(('2011/04/04', '2011/04/05')) obssum = rhessi.parse_obssumm_dbase_file(file[0]) assert obssum['filename'][0] == 'hsi_obssumm_20110401_043.fit' assert obssum['filename'][-1] == 'hsi_obssumm_20110430_029.fit' assert obssum['orb_st'][0] == 0 assert obssum['orb_st'][-1] == 0 assert obssum['orb_end'][0] == 0 assert obssum['orb_end'][-1] == 0 assert obssum['start_time'][0] == datetime(2011, 4, 1, 0, 0, 0) assert obssum['start_time'][-1] == datetime(2011, 4, 30, 0, 0, 0) assert obssum['end_time'][0] == datetime(2011, 4, 2, 0, 0, 0) assert obssum['end_time'][-1] == datetime(2011, 5, 1, 0, 0, 0) assert obssum['status_flag'][0] == 0 assert obssum['status_flag'][-1] == 0 assert obssum['npackets'][0] == 0 assert obssum['npackets'][-1] == 0