Skip to content

Commit

Permalink
add pytest for getargspec
Browse files Browse the repository at this point in the history
  • Loading branch information
Zep-Tepi committed Oct 24, 2024
1 parent 0f24167 commit f20d87c
Showing 1 changed file with 45 additions and 0 deletions.
45 changes: 45 additions & 0 deletions src/pymdgen/test_getargspec.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
import inspect
from functools import wraps
from pymdgen import getargspec # Import your __init__.py module


def test_getargspec_simple_function():
def my_function(a, b, c=10):
pass
expected_result = [
['a', 'b', 'c'],
None,
None,
[10,],
]
assert getargspec(my_function) == expected_result

def test_getargspec_args_kwargs():
def my_function(a, b, *args, **kwargs):
pass
expected_result = [
['a', 'b', 'args', 'kwargs'],
None,
None,
[],
]
assert getargspec(my_function) == expected_result

def test_getargspec_decorated_function():
def my_decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
return func(*args, **kwargs)
return wrapper

@my_decorator
def my_function(a, b, c=20):
pass

expected_result = [
['a', 'b', 'c'],
None,
None,
[20,],
]
assert getargspec(my_function) == expected_result

0 comments on commit f20d87c

Please sign in to comment.