You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.7 KiB
84 lines
2.7 KiB
6 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# pylint: disable=no-self-use, pointless-statement, missing-docstring, invalid-name, len-as-condition
|
||
|
|
|
||
|
|
from ..loose import call
|
||
|
|
|
||
|
|
|
||
|
|
def test_loose_function():
|
||
|
|
|
||
|
|
def func(v1, v2, v3=3, v4=4):
|
||
|
|
return v1 + v2 + v3 + v4
|
||
|
|
|
||
|
|
assert call(func, 1, 2) == func(1, 2)
|
||
|
|
assert call(func, 1, 2, 3, 5) == func(1, 2, 3, 5)
|
||
|
|
assert call(func, 1, 2, v3=4, v4=5) == func(1, 2, v3=4, v4=5)
|
||
|
|
assert call(func, 1, 2, 3, 4, 5) == func(1, 2, 3, 4)
|
||
|
|
assert call(func, 1, 2, 3, 4, more=5) == func(1, 2, 3, 4)
|
||
|
|
|
||
|
|
|
||
|
|
def test_loose_varargs_function():
|
||
|
|
def func(v1, v2, *args):
|
||
|
|
return v1 + v2 + args[0] if len(args) > 0 else 3 + args[1] if len(args) > 1 else 4
|
||
|
|
|
||
|
|
assert call(func, 1, 2) == func(1, 2)
|
||
|
|
assert call(func, 1, 2, 3, 5) == func(1, 2, 3, 5)
|
||
|
|
assert call(func, 1, 2, 3, 4, 5) == func(1, 2, 3, 4)
|
||
|
|
|
||
|
|
|
||
|
|
def test_loose_kwargs_function():
|
||
|
|
def func(v1, v2, **kwargs):
|
||
|
|
return v1 + v2 + kwargs.get('v3', 3) + kwargs.get('v4', 4)
|
||
|
|
|
||
|
|
assert call(func, v1=1, v2=2) == func(v1=1, v2=2)
|
||
|
|
assert call(func, v1=1, v2=2, v3=3, v4=5) == func(v1=1, v2=2, v3=3, v4=5)
|
||
|
|
|
||
|
|
|
||
|
|
def test_loose_class():
|
||
|
|
class Dummy(object):
|
||
|
|
def __init__(self, v1, v2, v3=3, v4=4):
|
||
|
|
self.v1 = v1
|
||
|
|
self.v2 = v2
|
||
|
|
self.v3 = v3
|
||
|
|
self.v4 = v4
|
||
|
|
|
||
|
|
def call(self):
|
||
|
|
return self.v1 + self.v2 + self.v3 + self.v4
|
||
|
|
|
||
|
|
assert call(Dummy, 1, 2).call() == Dummy(1, 2).call()
|
||
|
|
assert call(Dummy, 1, 2, 3, 5).call() == Dummy(1, 2, 3, 5).call()
|
||
|
|
assert call(Dummy, 1, 2, v3=4, v4=5).call() == Dummy(1, 2, v3=4, v4=5).call()
|
||
|
|
assert call(Dummy, 1, 2, 3, 4, 5).call() == Dummy(1, 2, 3, 4).call()
|
||
|
|
assert call(Dummy, 1, 2, 3, 4, more=5).call() == Dummy(1, 2, 3, 4).call()
|
||
|
|
|
||
|
|
|
||
|
|
def test_loose_varargs_class():
|
||
|
|
class Dummy(object):
|
||
|
|
def __init__(self, v1, v2, *args):
|
||
|
|
self.v1 = v1
|
||
|
|
self.v2 = v2
|
||
|
|
self.v3 = args[0] if len(args) > 0 else 3
|
||
|
|
self.v4 = args[1] if len(args) > 1 else 4
|
||
|
|
|
||
|
|
def call(self):
|
||
|
|
return self.v1 + self.v2 + self.v3 + self.v4
|
||
|
|
|
||
|
|
assert call(Dummy, 1, 2).call() == Dummy(1, 2).call()
|
||
|
|
assert call(Dummy, 1, 2, 3, 5).call() == Dummy(1, 2, 3, 5).call()
|
||
|
|
assert call(Dummy, 1, 2, 3, 4, 5).call() == Dummy(1, 2, 3, 4).call()
|
||
|
|
|
||
|
|
|
||
|
|
def test_loose_kwargs_class():
|
||
|
|
class Dummy(object):
|
||
|
|
def __init__(self, v1, v2, **kwargs):
|
||
|
|
self.v1 = v1
|
||
|
|
self.v2 = v2
|
||
|
|
self.v3 = kwargs.get('v3', 3)
|
||
|
|
self.v4 = kwargs.get('v4', 4)
|
||
|
|
|
||
|
|
def call(self):
|
||
|
|
return self.v1 + self.v2 + self.v3 + self.v4
|
||
|
|
|
||
|
|
assert call(Dummy, v1=1, v2=2).call() == Dummy(v1=1, v2=2).call()
|
||
|
|
assert call(Dummy, v1=1, v2=2, v3=3, v4=5).call() == Dummy(v1=1, v2=2, v3=3, v4=5).call()
|