from collections import OrderedDict
import numpy as np
import dqn

def test_agent():
    case = [1,0,1,0,0,0,0,0,0,0,0,0,0,0,0]
    event = 3
    process = [1,4,2,2,2,4,0,2,3,3,20,20,20]

    state = OrderedDict()
    state['case'] = np.asarray(case) 
    state['event'] = event
    state['process'] = np.asarray(process)
    print(state)

    print(dqn.deploy(state))

test_agent()