ev3_problem.py 文件源码

def test(args):
    '''Runs a single test, using a prevously generated individual'''
    global database
    global goal_positions
    pygame.init()
    random.seed()
    np.random.seed()
    if args.database is None:
        args.database = 'RobotTrainingData'
    database = Database(args.database)
    problem = EV3Problem(log_to_file=False)

    res_path = os.path.abspath(pkg_resources.resource_filename('resources.ev3', 'y_test.txt'))
    goal_positions = np.loadtxt(res_path)

    database.select(args.generation)
    chromosome = np.fromstring(database.load()['I' + str(args.individual)]).tolist()
    problem.robot.home()
    for goal in goal_positions:
        problem.run_test(goal, chromosome)