def test_create_display_table():
conf.max_width = 500
entries = [
DatabaseEntry(
id=1, source='SOHO', provider='SDAC', physobs='intensity',
fileid='/archive/soho/...',
observation_time_start=datetime(2001, 1, 1, 7, 0, 14),
observation_time_end=datetime(2001, 1, 1, 7, 0, 21),
instrument='EIT', size=259.0, wavemin=171.0,
wavemax=171.0, tags=[Tag('foo'), Tag('bar')]),
DatabaseEntry(
id=2, source='GONG', provider='NSO', physobs='LOS_velocity',
fileid='pptid=11010...',
observation_time_start=datetime(2010, 1, 1, 0, 59),
observation_time_end=datetime(2010, 1, 1, 1),
download_time=datetime(2014, 6, 15, 3, 42, 55, 123456),
instrument='Merged gong', size=944.0,
wavemin=6768.0, wavemax=6768.0, starred=True)]
columns = [
'id', 'source', 'provider', 'physobs', 'fileid', 'download_time',
'observation_time_start', 'instrument', 'size',
'wavemin', 'path', 'starred', 'tags']
table = _create_display_table(entries, columns)
filedir = os.path.dirname(os.path.realpath(__file__))
with open(os.path.join(filedir,'test_table.txt'), 'r') as f:
stored_table = f.read()
assert table.__str__().strip() == stored_table.strip()
conf.reset('max_width')
def test_create_display_table():
conf.max_width = 500
entries = [
DatabaseEntry(
id=1, source='SOHO', provider='SDAC', physobs='intensity',
fileid='/archive/soho/...',
observation_time_start=datetime(2001, 1, 1, 7, 0, 14),
observation_time_end=datetime(2001, 1, 1, 7, 0, 21),
instrument='EIT', size=259.0, wavemin=171.0,
wavemax=171.0, tags=[Tag('foo'), Tag('bar')]),
DatabaseEntry(
id=2, source='GONG', provider='NSO', physobs='LOS_velocity',
fileid='pptid=11010...',
observation_time_start=datetime(2010, 1, 1, 0, 59),
observation_time_end=datetime(2010, 1, 1, 1),
download_time=datetime(2014, 6, 15, 3, 42, 55, 123456),
instrument='Merged gong', size=944.0,
wavemin=6768.0, wavemax=6768.0, starred=True)]
columns = [
'id', 'source', 'provider', 'physobs', 'fileid', 'download_time',
'observation_time_start', 'instrument', 'size',
'wavemin', 'path', 'starred', 'tags']
table = _create_display_table(entries, columns)
filedir = os.path.dirname(os.path.realpath(__file__))
with open(os.path.join(filedir, 'test_table.txt'), 'r') as f:
stored_table = f.read()
assert table.__str__().strip() == stored_table.strip()
conf.reset('max_width')
def _repr_html_(self):
return _create_display_table(self)._repr_html_()
def __str__(self):
return _create_display_table(self).__str__()
def show_in_browser(self, columns=None, sort=False, jsviewer=True):
_create_display_table(self, columns, sort).show_in_browser(jsviewer)
def display_entries(self, columns=None, sort=False):
print(_create_display_table(self, columns, sort))
def test_create_display_table_missing_columns():
with pytest.raises(TypeError):
_create_display_table([DatabaseEntry()], [])
def test_create_display_table_empty_db():
with pytest.raises(TypeError):
_create_display_table(Database('sqlite:///'), ['id'])
def test_create_display_table_missing_entries():
with pytest.raises(TypeError):
_create_display_table([], ['some', 'columns'])
def show_in_browser(self, columns=None, sort=False, jsviewer=True):
_create_display_table(self, columns, sort).show_in_browser(jsviewer)
def display_entries(self, columns=None, sort=False):
print(_create_display_table(self, columns, sort))
def _repr_html_(self):
return _create_display_table(self)._repr_html_()
def __str__(self):
return _create_display_table(self).__str__()
def test_create_display_table_missing_columns():
with pytest.raises(TypeError):
_create_display_table([DatabaseEntry()], [])
def test_create_display_table_empty_db():
with pytest.raises(TypeError):
_create_display_table(Database('sqlite:///'), ['id'])
def test_create_display_table_missing_entries():
with pytest.raises(TypeError):
_create_display_table([], ['some', 'columns'])