def test_tuple(self):
     id_ = lambda x: x
     in_spec = api.Shape(())
     out_spec = api.spec((id_, id_, id_))(in_spec)
     tupl = api.init((id_, id_, id_))(self._seed, in_spec)
     self.assertTupleEqual(out_spec, (in_spec, ) * 3)
     self.assertTupleEqual(api.call(tupl, 1.), (1., 1., 1.))
 def test_list(self):
     func = lambda x: x + 1
     in_spec = api.Shape(())
     out_spec = api.spec([func, func, func])(in_spec)
     lst = api.init([func, func, func])(self._seed, in_spec)
     self.assertEqual(out_spec, in_spec)
     self.assertEqual(api.call(lst, 1.), 4.)
 def test_func(self):
     id_ = lambda x: x
     in_spec = api.Shape(())
     out_spec = api.spec(id_)(in_spec)
     func = api.init(id_)(self._seed, in_spec)
     self.assertEqual(out_spec, in_spec)
     self.assertEqual(api.call(func, 1.), 1.)
 def test_list_multiple_args(self):
     func1 = lambda x: (x, x + 1)
     func2 = lambda x, y: (y, x + 1)
     func3 = lambda x, y: (x + 1, y + 2, x + y)
     in_spec = api.Shape(())
     out_spec = api.spec([func1, func2, func3])(in_spec)
     lst = api.init([func1, func2, func3])(self._seed, in_spec)
     self.assertEqual(out_spec, (api.Shape(()), ) * 3)
     self.assertTupleEqual(api.call(lst, 1.), (3., 4., 4.))