def test_get_table(): table_set = PropertyMock() tables_result = [Mock() for _ in range(3)] tables_result[0].name = 'tab1' tables_result[1].name = 'tab2' tables_result[2].name = 'tab3' table_set.values.return_value = tables_result with patch('astroquery.cadc.core.pyvo.dal.TAPService', autospec=True) as tapservice_mock: tapservice_mock.return_value.tables = table_set cadc = Cadc() assert cadc.get_table('tab2').name == 'tab2' assert cadc.get_table('foo') is None
def test_get_table(monkeypatch): dummyTapHandler = DummyTapHandler() monkeypatch.setattr(cadc_core, 'get_access_url', get_access_url_mock) tap = Cadc(tap_plus_handler=dummyTapHandler) # default parameters parameters = {} parameters['table'] = 'table' parameters['verbose'] = False tap.get_table('table') dummyTapHandler.check_call('get_table', parameters) # test with parameters dummyTapHandler.reset() parameters = {} parameters['table'] = 'table' parameters['verbose'] = True tap.get_table('table', verbose=True) dummyTapHandler.check_call('get_table', parameters)
def test_list_tables(self): cadc = Cadc() table_names = cadc.get_tables(only_names=True) assert len(table_names) > 20 assert 'caom2.Observation' in table_names assert 'ivoa.ObsCore' in table_names assert 'tap_schema.tables' in table_names tables = cadc.get_tables() assert len(table_names) == len(tables) for table in tables: assert table.name in table_names table = cadc.get_table('caom2.Observation') assert 'caom2.Observation' == table.name