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))