File: test_resolver.py

package info (click to toggle)
python-datamodel-code-generator 0.33.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,324 kB
  • sloc: python: 19,560; makefile: 15
file content (18 lines) | stat: -rw-r--r-- 450 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
from __future__ import annotations

import pytest

from datamodel_code_generator.reference import FieldNameResolver


@pytest.mark.parametrize(
    ("name", "expected_resolved"),
    [
        ("3a", "field_3a"),
        ("$in", "field_in"),
        ("field", "field"),
    ],
)
def test_get_valid_field_name(name: str, expected_resolved: str) -> None:
    resolver = FieldNameResolver()
    assert expected_resolved == resolver.get_valid_name(name)