1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
import json
from uuid import UUID
import pytest
from pydantic import TypeAdapter, ValidationError
class TestAll:
def test_comma_sep_str(self):
from generalresearch.models.custom_types import AlphaNumStrSet
t = TypeAdapter(AlphaNumStrSet)
assert {"a", "b", "c"} == t.validate_python(["a", "b", "c"])
assert '"a,b,c"' == t.dump_json({"c", "b", "a"}).decode()
assert '""' == t.dump_json(set()).decode()
assert {"a", "b", "c"} == t.validate_json('"c,b,a"')
assert set() == t.validate_json('""')
with pytest.raises(ValidationError):
t.validate_python({"", "b", "a"})
with pytest.raises(ValidationError):
t.validate_python({""})
with pytest.raises(ValidationError):
t.validate_json('",b,a"')
def test_UUIDStrCoerce(self):
from generalresearch.models.custom_types import UUIDStrCoerce
t = TypeAdapter(UUIDStrCoerce)
uuid_str = "18e70590176e49c693b07682f3c112be"
assert uuid_str == t.validate_python("18e70590-176e-49c6-93b0-7682f3c112be")
assert uuid_str == t.validate_python(
UUID("18e70590-176e-49c6-93b0-7682f3c112be")
)
assert (
json.dumps(uuid_str)
== t.dump_json("18e70590176e49c693b07682f3c112be").decode()
)
assert uuid_str == t.validate_json('"18e70590-176e-49c6-93b0-7682f3c112be"')
|
