def test_load_data(self): dummy_handler = DummyTapHandler() tap = GaiaClass(dummy_handler, dummy_handler, show_server_messages=False) ids = "1,2,3,4" retrieval_type = "epoch_photometry" valid_data = True band = None format = "votable" verbose = True data_structure = "INDIVIDUAL" output_file = os.path.abspath("output_file") path_to_end_with = os.path.join("gaia", "test", "output_file") if not output_file.endswith(path_to_end_with): output_file = os.path.abspath(path_to_end_with) params_dict = {} params_dict['VALID_DATA'] = "true" params_dict['ID'] = ids params_dict['FORMAT'] = str(format) params_dict['RETRIEVAL_TYPE'] = str(retrieval_type) params_dict['DATA_STRUCTURE'] = str(data_structure) params_dict['USE_ZIP_ALWAYS'] = 'true' tap.load_data(ids=ids, retrieval_type=retrieval_type, valid_data=valid_data, band=band, format=format, verbose=verbose, output_file=output_file) parameters = {} parameters['params_dict'] = params_dict # Output file name contains a timestamp: cannot be verified of = dummy_handler._DummyTapHandler__parameters['output_file'] parameters['output_file'] = of parameters['verbose'] = verbose dummy_handler.check_call('load_data', parameters)
def test_load_data(self): dummyHandler = DummyTapHandler() tap = GaiaClass(dummyHandler, dummyHandler) ids = "1,2,3,4" retrieval_type = "epoch_photometry" valid_data = True band = None format = "votable" verbose = True params_dict = {} params_dict['VALID_DATA'] = "true" params_dict['ID'] = ids params_dict['FORMAT'] = str(format) params_dict['RETRIEVAL_TYPE'] = str(retrieval_type) tap.load_data(ids=ids, retrieval_type=retrieval_type, valid_data=valid_data, band=band, format=format, verbose=verbose) parameters = {} parameters['params_dict'] = params_dict parameters['output_file'] = None parameters['verbose'] = verbose dummyHandler.check_call('load_data', parameters) tap.load_data(ids=ids, retrieval_type=retrieval_type, valid_data=valid_data, band=band, format=format, verbose=verbose) dummyHandler.check_call('load_data', parameters)