def test_QueryResponse_build_table_defaults():
records = (MockQRRecord(), )
qr = vso.QueryResponse(records)
table = qr.build_table()
start_time_ = table['Start Time']
assert len(start_time_) == 1
assert start_time_[0] == 'None'
end_time_ = table['End Time']
assert len(end_time_) == 1
assert end_time_[0] == 'None'
type_ = table['Type'].data
assert len(type_) == 1
assert type_[0] == 'N/A'
# Check values we did set by default in 'MockQRRecord'
source_ = table['Source'].data
assert len(source_) == 1
assert source_[0] == str(va.Source('SOHO'))
instrument_ = table['Instrument'].data
assert len(instrument_) == 1
assert instrument_[0] == str(va.Instrument('aia'))
def test_QueryResponse_build_table_with_extent_type(mock_build_client):
"""
When explicitly suppling an 'Extent' only the 'type' is stored
in the built table.
"""
e_type = va.Extent(x=1.0, y=2.5, width=37, length=129.2, atype='CORONA')
qr = vso.QueryResponse((MockQRRecord(extent_type=e_type),))
table = qr.build_table()
extent = table['Extent Type'].data
assert len(extent) == 1
assert extent[0] == e_type.type
def test_QueryResponse_build_table_with_no_end_time(mock_build_client):
"""
Only the 'start' time is set, no 'end' time
"""
a_st = parse_time((2016, 2, 14, 8, 8, 12))
records = (MockQRRecord(start_time=a_st.strftime(va._TIMEFORMAT)),)
qr = vso.QueryResponse(records)
table = qr.build_table()
start_time_ = table['Start Time']
assert len(start_time_) == 1
assert start_time_[0].value == '2016-02-14 08:08:12.000'
def test_QueryResponse_build_table_with_no_start_time(mock_build_client):
"""
Only the 'end' time set, no 'start' time
"""
a_st = parse_time((2016, 2, 14, 8, 8, 12))
records = (MockQRRecord(end_time=a_st.strftime(va._TIMEFORMAT)),)
qr = vso.QueryResponse(records)
table = qr.build_table()
# 'End Time' is valid, there is no 'Start Time' in the table
assert 'Start Time' not in table.columns
end_time_ = table['End Time']
assert len(end_time_) == 1
assert end_time_[0].value == '2016-02-14 08:08:12.000'
def test_QueryResponse_build_table_with_no_end_time():
"""
Only the 'start' time is set, no 'end' time
"""
a_st = datetime.datetime(2016, 2, 14, 8, 8, 12)
records = (MockQRRecord(start_time=a_st.strftime(va.TIMEFORMAT)), )
qr = vso.QueryResponse(records)
table = qr.build_table()
start_time_ = table['Start Time']
assert len(start_time_) == 1
assert start_time_[0] == '2016-02-14 08:08:12'
end_time_ = table['End Time']
assert len(end_time_) == 1
assert end_time_[0] == 'None'
def test_QueryResponse_build_table_defaults(mock_build_client):
records = (MockQRRecord(),)
qr = vso.QueryResponse(records)
table = qr.build_table()
# These are the only None values in the table.
source_ = table['Source'].data
assert len(source_) == 1
assert source_[0].value == 'SOHO'
instrument_ = table['Instrument'].data
assert len(instrument_) == 1
assert instrument_[0].value == 'aia'
size_ = table['Size'].data
assert len(size_) == 1
assert size_[0] == 0.0
def test_QueryResponse_build_table_with_no_start_time():
"""
Only the 'end' time set, no 'start' time
"""
a_st = parse_time((2016, 2, 14, 8, 8, 12))
records = (MockQRRecord(end_time=a_st.strftime(va.TIMEFORMAT)), )
qr = vso.QueryResponse(records)
table = qr.build_table()
start_time_ = table['Start Time']
assert len(start_time_) == 1
assert start_time_[0] == 'None'
# Even though 'End Time' is valid, there is no 'Start Time'
# marks as 'N/A'
end_time_ = table['End Time']
assert len(end_time_) == 1
assert end_time_[0] == 'N/A'