def test_arg_defaults_with_vararg(self): def test_fn(a, b=1, *c): # pylint: disable=keyword-arg-before-vararg return a, b, c node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) self.assertTransformedFirstLineIs(node, 'def test_fn(a, b=None, *c):')
def test_no_defaults(self): def test_fn(a, b, *c, **e): return a, b, c, e node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) self.assertTransformedFirstLineIs(node, 'def test_fn(a, b, *c, **e):')
def test_no_args(self): def test_fn(): pass node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) self.assertTransformedFirstLineIs(node, 'def test_fn():')
def test_arg_defaults_with_vararg(self): def test_fn(a, b=1, *c): # pylint: disable=keyword-arg-before-vararg return a, b, c node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) self.assertTransformedFirstLineIs(node, 'def test_fn(a, b=None, *c):')
def test_arg_defaults(self): def test_fn(a, b=1, c=2): return a, b, c node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) self.assertTransformedFirstLineIs(node, 'def test_fn(a, b=None, c=None):')
def test_no_defaults(self): def test_fn(a, b, *c, **e): return a, b, c, e node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) self.assertTransformedFirstLineIs(node, 'def test_fn(a, b, *c, **e):')
def test_no_args(self): def test_fn(): pass node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) self.assertTransformedFirstLineIs(node, 'def test_fn():')
def test_arg_defaults_ignores_inner_lambda(self): def test_fn(): return (lambda x=7: x)() node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) with self.converted(test_fn, arg_defaults, {}) as result: self.assertEqual(test_fn(), result.test_fn())
def test_arg_defaults_ignores_inner_lambda(self): def test_fn(): return (lambda x=7: x)() node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) with self.converted(test_fn, arg_defaults, {}) as result: self.assertEqual(test_fn(), result.test_fn())
def test_arg_defaults_ignores_inner_function_returned(self): def test_fn(): def inner_fn(a=3): return a return inner_fn node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) with self.converted(test_fn, arg_defaults, {}) as result: self.assertEqual(test_fn()(), result.test_fn()())
def test_arg_defaults_ignores_inner_function_returned(self): def test_fn(): def inner_fn(a=3): return a return inner_fn node, ctx = self.prepare(test_fn, {}) node = arg_defaults.transform(node, ctx) with self.converted(test_fn, arg_defaults, {}) as result: self.assertEqual(test_fn()(), result.test_fn()())