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):')
Esempio n. 5
0
  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()())
Esempio n. 11
0
  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()())