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
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