Пример #1
0
    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)