def test_get_tables(): # default parameters table_set = PropertyMock() table_set.keys.return_value = ['table1', 'table2'] table_set.values.return_value = ['tab1val', 'tab2val', 'tab3val'] with patch('astroquery.cadc.core.pyvo.dal.TAPService', autospec=True) as tapservice_mock: tapservice_mock.return_value.tables = table_set cadc = Cadc() assert len(cadc.get_tables(only_names=True)) == 2 assert len(cadc.get_tables()) == 3
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
def test_get_tables(monkeypatch): dummyTapHandler = DummyTapHandler() monkeypatch.setattr(cadc_core, 'get_access_url', get_access_url_mock) tap = Cadc(tap_plus_handler=dummyTapHandler) # default parameters parameters = {} parameters['only_names'] = False parameters['verbose'] = False tap.get_tables() dummyTapHandler.check_call('get_tables', parameters) # test with parameters dummyTapHandler.reset() parameters = {} parameters['only_names'] = True parameters['verbose'] = True tap.get_tables(True, True) dummyTapHandler.check_call('get_tables', parameters)