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
|
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
|
