refactor: update flaky tests in test_schema.py (#3933)
* Fix flaky tests in `test_schema.py` by using sets for type comparisons - Updated `post_process_type` function assertions to use sets for more reliable type comparisons. - Adjusted imports for better code organization. * Fix import order in test_database.py * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
8facb67944
commit
706f9a0277
2 changed files with 8 additions and 8 deletions
|
|
@ -2,7 +2,6 @@ import json
|
|||
from collections import namedtuple
|
||||
from uuid import UUID, uuid4
|
||||
|
||||
from langflow.services.database.models.folder.model import FolderCreate
|
||||
import orjson
|
||||
import pytest
|
||||
from fastapi.testclient import TestClient
|
||||
|
|
@ -13,6 +12,7 @@ from langflow.graph.utils import log_transaction, log_vertex_build
|
|||
from langflow.initial_setup.setup import load_flows_from_directory, load_starter_projects
|
||||
from langflow.services.database.models.base import orjson_dumps
|
||||
from langflow.services.database.models.flow import Flow, FlowCreate, FlowUpdate
|
||||
from langflow.services.database.models.folder.model import FolderCreate
|
||||
from langflow.services.database.utils import session_getter
|
||||
from langflow.services.deps import get_db_service
|
||||
|
||||
|
|
|
|||
|
|
@ -1,5 +1,5 @@
|
|||
from typing import Union
|
||||
from collections.abc import Sequence
|
||||
from typing import Union
|
||||
|
||||
import pytest
|
||||
from pydantic import ValidationError
|
||||
|
|
@ -40,11 +40,11 @@ class TestInput:
|
|||
assert input_obj.field_type == "int"
|
||||
|
||||
def test_post_process_type_function(self):
|
||||
assert post_process_type(int) == [int]
|
||||
assert post_process_type(list[int]) == [int]
|
||||
assert post_process_type(Union[int, str]) == [int, str]
|
||||
assert post_process_type(Union[int, Sequence[str]]) == [int, str]
|
||||
assert post_process_type(Union[int, Sequence[int]]) == [int]
|
||||
assert set(post_process_type(int)) == {int}
|
||||
assert set(post_process_type(list[int])) == {int}
|
||||
assert set(post_process_type(Union[int, str])) == {int, str}
|
||||
assert set(post_process_type(Union[int, Sequence[str]])) == {int, str}
|
||||
assert set(post_process_type(Union[int, Sequence[int]])) == {int}
|
||||
|
||||
def test_input_to_dict(self):
|
||||
input_obj = Input(field_type="str")
|
||||
|
|
@ -110,7 +110,7 @@ class TestPostProcessType:
|
|||
assert post_process_type(list[int]) == [int]
|
||||
|
||||
def test_union_type(self):
|
||||
assert post_process_type(Union[int, str]) == [int, str]
|
||||
assert set(post_process_type(Union[int, str])) == {int, str}
|
||||
|
||||
def test_custom_type(self):
|
||||
class CustomType:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue