def test_pixels(self): """Runs reference pixel and mean-square comparisons on extracts from randomly generated maps""" print("Testing reference pixels...") results,rname = ptests.get_extraction_test_results(TEST_DIR+"/tests.yml") cresults = pickle.load(open(DATA_PREFIX+"%s.pkl" % rname,'rb')) assert sorted(results.keys())==sorted(cresults.keys()) for g in results.keys(): assert sorted(results[g].keys())==sorted(cresults[g].keys()) for s in results[g].keys(): assert sorted(results[g][s].keys())==sorted(cresults[g][s].keys()) for e in results[g][s].keys(): assert np.all(np.isclose(results[g][s][e],cresults[g][s][e]))
import pixel_tests as ptests import pickle import os import argparse # Parse command line parser = argparse.ArgumentParser(description='Generate test results.') parser.add_argument("-y", "--yaml", type=str, default=None,help="Test YAML file.") parser.add_argument("-o", "--output", type=str,default=None,help='Name of output pkl file.') args = parser.parse_args() i = 0 if args.yaml is None: yaml_file = "" while not(os.path.isfile(yaml_file)) or i==0: if i>0: print("File does not exist.") yaml_file = input("Enter name of yaml file to generate test results from: ") i += 1 else: yaml_file = args.yaml if args.output is None: pkl_file = input("Enter name of pickle file to save test results to (exclude extension): ") else: pkl_file = args.output results,_ = ptests.get_extraction_test_results(yaml_file) pickle.dump(results,open("data/%s.pkl" % pkl_file,'wb'),protocol=2)