31 lines
721 B
Python
31 lines
721 B
Python
|
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from suou import dei_args
|
||
|
|
|
||
|
|
class TestDei(unittest.TestCase):
|
||
|
|
def setUp(self) -> None:
|
||
|
|
...
|
||
|
|
|
||
|
|
def tearDown(self) -> None:
|
||
|
|
...
|
||
|
|
|
||
|
|
def test_dei_args(self):
|
||
|
|
|
||
|
|
def func_a(*, a: int, b: int):
|
||
|
|
"""Trivial function for the purpose of the test"""
|
||
|
|
return a - b == 0
|
||
|
|
|
||
|
|
func_b = dei_args(c='a', d='b')(func_a)
|
||
|
|
|
||
|
|
with self.assertRaises(TypeError):
|
||
|
|
func_a(c=1, b=2)
|
||
|
|
|
||
|
|
self.assertTrue(func_a(a=1, b=1))
|
||
|
|
self.assertFalse(func_b(c=1, b=2))
|
||
|
|
self.assertFalse(func_b(a=1, d=2))
|
||
|
|
self.assertFalse(func_b(a=1, b=2))
|
||
|
|
|
||
|
|
with self.assertRaises(TypeError):
|
||
|
|
func_b(c=1, b="a")
|