def test_read_csv(): assert read_csv('tests/fixtures/framework_results.csv') == [ ['123', 'Supplier Name', 'pass'], ['234', 'Supplier Name', 'fail'], ['345', 'Supplier Name', ' PASS'], ['456', 'Supplier Name', ' FAIL'], ['567', 'Supplier Name', ' Yes'], ['Company Name', 'Supplier Name', 'pass'], ['678', 'Supplier Name', 'PasS'], ]
Example: ./generate-framework-agreement-data.py lots.csv declarations.csv framework-outputs localhost:5000 myToken """ import sys sys.path.insert(0, '.') from docopt import docopt from dmscripts.generate_framework_agreement_data import check_lots_csv, read_csv, \ check_declarations_csv, build_framework_agreements from dmutils.apiclient import DataAPIClient if __name__ == '__main__': arguments = docopt(__doc__) supplier_lots = read_csv(arguments['<lots_file>']) supplier_declarations = read_csv(arguments['<declaration_file>']) # Remove first column titles row, if it exists if supplier_lots[0][0] == "Digital Marketplace ID": del supplier_lots[0] if supplier_declarations[0][0] == "Digital Marketplace ID": del supplier_declarations[0] lot_check = check_lots_csv(supplier_lots) if not lot_check[0]: sys.exit("Lots CSV is invalid: {}".format(lot_check[1])) declaration_check = check_declarations_csv(supplier_declarations) if not declaration_check[0]: