Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions .github/workflows/tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -107,16 +107,15 @@ jobs:
run: |
pip install .[coverage]
pip install "pydantic<2"
pytest --cov --cov-report=term --cov-report=xml --junit-xml=junit.xml jsonargparse_tests/test_dataclass_like.py
pytest --cov --cov-report=term --cov-report=xml --junit-xml=junit.xml jsonargparse_tests/test_pydantic.py
mv coverage.xml coverage_pydantic1.xml
mv junit.xml junit_pydantic1.xml
- name: with pydantic>=2
run: |
sed -i "s|import pydantic|import pydantic.v1 as pydantic|" jsonargparse_tests/test_dataclass_like.py
sed -i "s|^annotated = .*|annotated = False|" jsonargparse_tests/test_dataclass_like.py
sed -i "s|test_pydantic_types|_test_pydantic_types|" jsonargparse_tests/test_dataclass_like.py
sed -i "s|import pydantic|import pydantic.v1 as pydantic|" jsonargparse_tests/test_pydantic.py
sed -i "s|^annotated = .*|annotated = False|" jsonargparse_tests/test_pydantic.py
pip install "pydantic>=2"
pytest --cov --cov-report=term --cov-report=xml --junit-xml=junit.xml jsonargparse_tests/test_dataclass_like.py
pytest --cov --cov-report=term --cov-report=xml --junit-xml=junit.xml jsonargparse_tests/test_pydantic.py
mv coverage.xml coverage_pydantic2.xml
mv junit.xml junit_pydantic2.xml
- uses: actions/upload-artifact@v4
Expand Down
89 changes: 89 additions & 0 deletions jsonargparse_tests/test_attrs.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,89 @@
from __future__ import annotations

from typing import List

import pytest

from jsonargparse import Namespace
from jsonargparse._optionals import attrs_support
from jsonargparse_tests.conftest import get_parser_help

if attrs_support:
import attrs

@attrs.define
class AttrsData:
p1: float
p2: str = "-"

@attrs.define
class AttrsSubData(AttrsData):
p3: int = 3

@attrs.define
class AttrsFieldFactory:
p1: List[str] = attrs.field(factory=lambda: ["one", "two"])

@attrs.define
class AttrsFieldInitFalse:
p1: dict = attrs.field(init=False)

def __attrs_post_init__(self):
self.p1 = {}

@attrs.define
class AttrsSubField:
p1: str = "-"
p2: int = 0

@attrs.define
class AttrsWithNestedDefaultDataclass:
p1: float
subfield: AttrsSubField = attrs.field(factory=AttrsSubField)

@attrs.define
class AttrsWithNestedDataclassNoDefault:
p1: float
subfield: AttrsSubField


@pytest.mark.skipif(not attrs_support, reason="attrs package is required")
class TestAttrs:
def test_define(self, parser):
parser.add_argument("--data", type=AttrsData)
defaults = parser.get_defaults()
assert Namespace(p1=None, p2="-") == defaults.data
cfg = parser.parse_args(["--data.p1=0.2", "--data.p2=x"])
assert Namespace(p1=0.2, p2="x") == cfg.data

def test_subclass(self, parser):
parser.add_argument("--data", type=AttrsSubData)
defaults = parser.get_defaults()
assert Namespace(p1=None, p2="-", p3=3) == defaults.data

def test_field_factory(self, parser):
parser.add_argument("--data", type=AttrsFieldFactory)
cfg1 = parser.parse_args([])
cfg2 = parser.parse_args([])
assert cfg1.data.p1 == ["one", "two"]
assert cfg1.data.p1 == cfg2.data.p1
assert cfg1.data.p1 is not cfg2.data.p1

def test_field_init_false(self, parser):
parser.add_argument("--data", type=AttrsFieldInitFalse)
cfg = parser.parse_args([])
help_str = get_parser_help(parser)
assert "--data.p1" not in help_str
assert cfg == Namespace()
init = parser.instantiate_classes(cfg)
assert init.data.p1 == {}

def test_nested_with_default(self, parser):
parser.add_argument("--data", type=AttrsWithNestedDefaultDataclass)
cfg = parser.parse_args(["--data.p1=1.23"])
assert cfg.data == Namespace(p1=1.23, subfield=Namespace(p1="-", p2=0))

def test_nested_without_default(self, parser):
parser.add_argument("--data", type=AttrsWithNestedDataclassNoDefault)
cfg = parser.parse_args(["--data.p1=1.23"])
assert cfg.data == Namespace(p1=1.23, subfield=Namespace(p1="-", p2=0))
2 changes: 1 addition & 1 deletion jsonargparse_tests/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def test_parse_object_simple(parser):
pytest.raises(ArgumentError, lambda: parser.parse_object({"undefined": True}))


def test_parse_object_config(parser):
def test_parse_object_config(parser, tmp_cwd):
parser.add_argument("--cfg", action="config")
parser.add_argument("--a", type=int)
parser.add_argument("--b", type=int)
Expand Down
Loading