Method test_loops

caffe2/python/net_builder_test.py:157–165

(self)

            self.assertEqual(2, result.fetch())

    def test_loops(self):
        with Task() as task:
            out_actual = self._actual_loop()
        with LocalSession() as session:
            session.run(task)
            expected = self._expected_loop()
            actual = [o.fetch() for o in out_actual]
            for e, a in zip(expected, actual):
                self.assertEqual(e, a)

    def test_setup(self):
        with Task() as task: