def test_load_and_save(): env = gym.make('CartPole-v0') M = Model(env, build_graph, build_update_feed_dict) M.update([0, 0, 0, 0], 2) weights = M.G._session.run(M['vars']) M.save('test_load_and_save_model') M.update([0, 0, 0, 0], 2) del M M = Model.load(path='test_load_and_save_model') weights2 = M.G._session.run(M['vars']) M.build_update_feed([0, 1, 0, 1], 2) M.update([0, 0, 0, 0], 2) shutil.rmtree('test_load_and_save_model') assert np.allclose(weights, weights2)
def test_update_model(): env = gym.make('CartPole-v0') M = Model(env, build_graph, build_update_feed_dict) loss = M.update([0, 0, 0, 0], 2) assert loss[0] == [4.]