Ejemplo n.º 1
0
 def test_unpack_args_from_tuple_type(self, tuple_with_args, expected_args,
                                      expected_kwargs):
   args, kwargs = function_utils.unpack_args_from_tuple(tuple_with_args)
   self.assertEqual(len(args), len(expected_args))
   for idx, arg in enumerate(args):
     self.assertTrue(
         arg.is_equivalent_to(computation_types.to_type(expected_args[idx])))
   self.assertEqual(set(kwargs.keys()), set(expected_kwargs.keys()))
   for k, v in kwargs.items():
     self.assertTrue(
         v.is_equivalent_to(computation_types.to_type(expected_kwargs[k])))
Ejemplo n.º 2
0
 def test_unpack_args_from_anonymous_tuple(self, tuple_with_args,
                                           expected_args, expected_kwargs):
     self.assertEqual(
         function_utils.unpack_args_from_tuple(tuple_with_args),
         (expected_args, expected_kwargs))