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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
from __future__ import annotations
import pytest
from textual_fastdatatable.backend import DataTableBackend
def test_column_content_widths(backend: DataTableBackend) -> None:
assert backend.column_content_widths == [1, 8, 6]
def test_get_row_at(backend: DataTableBackend) -> None:
assert backend.get_row_at(0) == [1, "a", "foo"]
assert backend.get_row_at(4) == [5, "asdfasdf", "foofoo"]
with pytest.raises(IndexError):
backend.get_row_at(10)
with pytest.raises(IndexError):
backend.get_row_at(-1)
def test_get_column_at(backend: DataTableBackend) -> None:
assert backend.get_column_at(0) == [1, 2, 3, 4, 5]
assert backend.get_column_at(2) == ["foo", "bar", "baz", "qux", "foofoo"]
with pytest.raises(IndexError):
backend.get_column_at(10)
def test_get_cell_at(backend: DataTableBackend) -> None:
assert backend.get_cell_at(0, 0) == 1
assert backend.get_cell_at(4, 1) == "asdfasdf"
with pytest.raises(IndexError):
backend.get_cell_at(10, 0)
with pytest.raises(IndexError):
backend.get_cell_at(0, 10)
def test_append_column(backend: DataTableBackend) -> None:
original_table = backend.data
backend.append_column("new")
assert backend.column_count == 4
assert backend.row_count == 5
assert backend.get_column_at(3) == [None] * backend.row_count
backend.append_column("def", default="zzz")
assert backend.column_count == 5
assert backend.row_count == 5
assert backend.get_column_at(4) == ["zzz"] * backend.row_count
assert backend.data.select(["first column", "two", "three"]).equals(original_table)
def test_append_rows(backend: DataTableBackend) -> None:
original_table = backend.data
backend.append_rows([(6, "w", "x"), (7, "y", "z")])
assert backend.column_count == 3
assert backend.row_count == 7
assert backend.column_content_widths == [1, 8, 6]
backend.append_rows([(999, "w" * 12, "x" * 15)])
assert backend.column_count == 3
assert backend.row_count == 8
assert backend.column_content_widths == [3, 12, 15]
assert backend.data.slice(0, 5).equals(original_table)
def test_drop_row(backend: DataTableBackend) -> None:
backend.drop_row(0)
assert backend.row_count == 4
assert backend.column_count == 3
assert backend.column_content_widths == [1, 8, 6]
backend.drop_row(3)
assert backend.row_count == 3
assert backend.column_count == 3
assert backend.column_content_widths == [1, 1, 3]
with pytest.raises(IndexError):
backend.drop_row(3)
def test_update_cell(backend: DataTableBackend) -> None:
backend.update_cell(0, 0, 0)
assert backend.get_column_at(0) == [0, 2, 3, 4, 5]
assert backend.row_count == 5
assert backend.column_count == 3
assert backend.column_content_widths == [1, 8, 6]
backend.update_cell(3, 1, "z" * 50)
assert backend.get_row_at(3) == [4, "z" * 50, "qux"]
assert backend.row_count == 5
assert backend.column_count == 3
assert backend.column_content_widths == [1, 50, 6]
def test_sort(backend: DataTableBackend) -> None:
original_table = backend.data
original_col_one = list(backend.get_column_at(0)).copy()
original_col_two = list(backend.get_column_at(1)).copy()
backend.sort(by="two")
assert backend.get_column_at(0) != original_col_one
assert backend.get_column_at(1) == sorted(original_col_two)
backend.sort(by=[("two", "descending")])
assert backend.get_column_at(0) != original_col_one
assert backend.get_column_at(1) == sorted(original_col_two, reverse=True)
backend.sort(by=[("first column", "ascending")])
assert backend.data.equals(original_table)
|
