test.py

from time import time

from env import Env_tsp
from config import Config, load_pkl, pkl_parser
from search import sampling, active_search


def search_tour(cfg: Config, env: Env_tsp) -> None:
    test_input = env.get_nodes(cfg.seed)

    # random
    print('generate random tour...')
    random_tour = env.get_random_tour()
    env.show(test_input, random_tour)

    # simplest way
    print('sampling ...')
    t1 = time()
    pred_tour = sampling(cfg, env, test_input)
    t2 = time()
    print('%dmin %1.2fsec\n' % ((t2 - t1) // 60, (t2 - t1) % 60))
    env.show(test_input, pred_tour)

    # active search, update parameters during test
    print('active search ...')
    t1 = time()
    pred_tour = active_search(cfg, env, test_input)
    t2 = time()
    print('%dmin %1.2fsec\n' % ((t2 - t1) // 60, (t2 - t1) % 60))
    env.show(test_input, pred_tour)

    '''
    # optimal solution, it takes time
    print('generate optimal solution ...')
    t1 = time()
    optimal_tour = env.get_optimal_tour(test_input)
    t2 = time()
    print('%dmin %1.2fsec\n' % ((t2 - t1) // 60, (t2 - t1) % 60))
    env.show(test_input, optimal_tour)
    '''


if __name__ == '__main__':
    cfg = load_pkl(pkl_parser().path)
    env = Env_tsp(cfg)

    # inputs = env.stack_nodes()
    # ~ tours = env.stack_random_tours()
    # ~ l = env.stack_l(inputs, tours)

    # ~ nodes = env.get_nodes(cfg.seed)
    # random_tour = env.get_random_tour()
    # ~ env.show(nodes, random_tour)

    # ~ env.show(inputs[0], random_tour)
    # ~ inputs = env.shuffle_index(inputs)
    # env.show(inputs[0], random_tour)

    # inputs = env.stack_nodes()
    # random_tour = env.get_random_tour()
    # env.show(inputs[0], random_tour)

    if cfg.mode == 'test':
        search_tour(cfg, env)

    else:
        raise NotImplementedError('test only, specify test pkl file')