5411889c97bb6de072d9b6aeab8b6bb352579de7,tf_agents/policies/async_policy_saver_test.py,AsyncPolicySaverTest,testClose,#AsyncPolicySaverTest#,80

Before Change

    saver.save_checkpoint.assert_has_calls([mock.call(path1), mock.call(path2)])

  def testClose(self):
    saver = mock.MagicMock()
    async_saver = async_policy_saver.AsyncPolicySaver(saver)
    path = os.path.join(self.get_temp_dir(), "save_model")

    self.evaluate(tf.compat.v1.global_variables_initializer())

After Change

    saver.save_checkpoint.assert_has_calls([mock.call(path1), mock.call(path2)])

  def testClose(self):
    saver = mock.create_autospec(policy_saver.PolicySaver, instance=True)
    async_saver = async_policy_saver.AsyncPolicySaver(saver)
    path = os.path.join(self.get_temp_dir(), "save_model")

    self.evaluate(tf.compat.v1.global_variables_initializer())

Instances