35 lines
1 KiB
Python
35 lines
1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Sequence, Type
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from textual_fastdatatable.backend import ArrowBackend, DataTableBackend, PolarsBackend
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def pydict() -> dict[str, Sequence[str | int]]:
|
||
|
|
return {
|
||
|
|
"first column": [1, 2, 3, 4, 5],
|
||
|
|
"two": ["a", "b", "c", "d", "asdfasdf"],
|
||
|
|
"three": ["foo", "bar", "baz", "qux", "foofoo"],
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def records(pydict: dict[str, Sequence[str | int]]) -> list[tuple[str | int, ...]]:
|
||
|
|
header = tuple(pydict.keys())
|
||
|
|
cols = list(pydict.values())
|
||
|
|
num_rows = len(cols[0])
|
||
|
|
data = [tuple([col[i] for col in cols]) for i in range(num_rows)]
|
||
|
|
return [header, *data]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(params=[ArrowBackend, PolarsBackend])
|
||
|
|
def backend(
|
||
|
|
request: Type[pytest.FixtureRequest], pydict: dict[str, Sequence[str | int]]
|
||
|
|
) -> DataTableBackend:
|
||
|
|
backend_cls = request.param
|
||
|
|
assert issubclass(backend_cls, (ArrowBackend, PolarsBackend))
|
||
|
|
backend: ArrowBackend | PolarsBackend = backend_cls.from_pydict(pydict)
|
||
|
|
return backend
|