diff --git a/aquerytest/aquerytest.py b/aquerytest/aquerytest.py index d805a3c..fd53781 100644 --- a/aquerytest/aquerytest.py +++ b/aquerytest/aquerytest.py @@ -1,4 +1,4 @@ -from astroquery.simbad import Simbad +import time __all__ = ['AQueryTest'] @@ -8,16 +8,27 @@ def __init__(self): pass def run(self): + from astroquery.simbad import Simbad + Simbad.add_votable_fields('flux(U)', 'flux(B)', 'flux(V)', 'flux(R)', 'flux(I)', 'flux(J)', 'sptype', 'parallax', 'pm', 'z_value') + print('Running first query...') table = Simbad.query_object('3C273') + print(table) if table is None: raise RuntimeError("Could not query.") + sleep = 600 + print(f'Sleeping for {sleep}s...') + time.sleep(sleep) + table = Simbad.query_object('3C273') + print('Success after a long sleep!') + return table def run_fail(self): + from astroquery.simbad import Simbad Simbad.add_votable_fields('flux(U)', 'flux(B)', 'flux(V)', 'flux(R)', 'flux(I)', 'flux(J)', 'sptype', 'parallax', 'pm', 'z_value') table = Simbad.query_object('BLAH') diff --git a/tests/test_aquerytest.py b/tests/test_aquerytest.py index 010fcc4..f6fe909 100644 --- a/tests/test_aquerytest.py +++ b/tests/test_aquerytest.py @@ -1,6 +1,5 @@ import aquerytest import pytest -import time def test_aquerytest(): @@ -12,7 +11,5 @@ def test_aquerytest(): with pytest.raises(RuntimeError): _ = tester.run_fail() - time.sleep(600) - table = tester.run() - print(table) + assert table is not None