コード例 #1
0
    def test_eval(self):
        root = Metric('test')
        root.eval = Mock()
        leaf = Metric('test')
        leaf.eval = Mock()

        tree = MetricTree(root)
        tree.add_child(leaf)

        tree.eval()
        root.eval.assert_called_once()
        leaf.eval.assert_called_once()
コード例 #2
0
ファイル: test_metrics.py プロジェクト: xinkez/torchbearer
    def test_eval(self):
        root = Metric('test')
        root.eval = Mock()
        leaf = Metric('test')
        leaf.eval = Mock()

        tree = MetricTree(root)
        tree.add_child(leaf)

        tree.eval()

        self.assertEqual(root.eval.call_count, 1)
        self.assertEqual(leaf.eval.call_count, 1)
コード例 #3
0
 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()
コード例 #4
0
ファイル: test_metrics.py プロジェクト: xinkez/torchbearer
 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)