def test_train(self): my_mock = Metric('test') my_mock.train = Mock(return_value=None) metric = MetricList([my_mock]) metric.train() my_mock.train.assert_called_once()
def test_eval(self): my_mock = Metric('test') my_mock.eval = Mock(return_value=None) metric = MetricList([my_mock]) metric.eval() my_mock.eval.assert_called_once()
def test_reset(self): my_mock = Metric('test') my_mock.reset = Mock(return_value=None) metric = MetricList([my_mock]) metric.reset({'state': -1}) my_mock.reset.assert_called_once_with({'state': -1})
def test_eval(self): my_mock = Metric('test') my_mock.eval = Mock(return_value=None) metric = MetricList([my_mock]) metric.eval() self.assertEqual(my_mock.eval.call_count, 1)
def test_train(self): my_mock = Metric('test') my_mock.train = Mock(return_value=None) metric = MetricList([my_mock]) metric.train() self.assertEqual(my_mock.train.call_count, 1)
def setUp(self): self._metric = Metric('test') self._mean = RunningMean('test') self._cache = [1.0, 1.5, 2.0] self._target = 1.5
def test_string(self): root = Metric('test') tree = MetricTree(root) self.assertEqual(str(root), str(tree))