def _test_strip_iterable(self, before, expected_after):
th = decorators.IOTypeHints(
input_types=None, output_types=((before, ), {}), origin=[])
after = th.strip_iterable()
self.assertEqual(((expected_after, ), {}), after.output_types)
def test_strip_iterable_not_simple_output_noop(self):
th = decorators.IOTypeHints(
input_types=None, output_types=((int, str), {}), origin=[])
th = th.strip_iterable()
self.assertEqual(((int, str), {}), th.output_types)
def default_type_hints(self):
return decorators.IOTypeHints(
input_types=((float, ), {}), output_types=((str, ), {}), origin=[])
def _test_strip_iterable(self, before, expected_after):
after = decorators.IOTypeHints(
output_types=((before,), {})).strip_iterable()
self.assertEqual(((expected_after, ), {}), after.output_types)
def default_type_hints(self):
return decorators.IOTypeHints(
input_types=((float, ), {}))
def default_type_hints(self):
return decorators.IOTypeHints(
input_types=((int, str), {}),
output_types=((int, ), {}))
