mirror of https://github.com/Textualize/rich.git
58 lines
3.3 KiB
Python
58 lines
3.3 KiB
Python
|
import io
|
||
|
|
|
||
|
|
from rich.columns import Columns
|
||
|
|
from rich.console import Console
|
||
|
|
|
||
|
|
COLUMN_DATA = [
|
||
|
|
"Ursus americanus",
|
||
|
|
"American buffalo",
|
||
|
|
"Bison bison",
|
||
|
|
"American crow",
|
||
|
|
"Corvus brachyrhynchos",
|
||
|
|
"American marten",
|
||
|
|
"Martes americana",
|
||
|
|
"American racer",
|
||
|
|
"Coluber constrictor",
|
||
|
|
"American woodcock",
|
||
|
|
"Scolopax minor",
|
||
|
|
"Anaconda (unidentified)",
|
||
|
|
"Eunectes sp.",
|
||
|
|
"Andean goose",
|
||
|
|
"Chloephaga melanoptera",
|
||
|
|
"Ant",
|
||
|
|
"Anteater, australian spiny",
|
||
|
|
"Tachyglossus aculeatus",
|
||
|
|
"Anteater, giant",
|
||
|
|
"Myrmecophaga tridactyla",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def render():
|
||
|
|
console = Console(file=io.StringIO(), width=100)
|
||
|
|
columns = Columns(COLUMN_DATA)
|
||
|
|
console.print(columns)
|
||
|
|
console.print()
|
||
|
|
columns.column_first = True
|
||
|
|
console.print(columns)
|
||
|
|
console.print()
|
||
|
|
columns.right_to_left = True
|
||
|
|
console.print(columns)
|
||
|
|
console.print()
|
||
|
|
columns.equal = True
|
||
|
|
columns.expand = True
|
||
|
|
console.print(columns)
|
||
|
|
console.print()
|
||
|
|
render_result = console.file.getvalue()
|
||
|
|
return render_result
|
||
|
|
|
||
|
|
|
||
|
|
def test_render():
|
||
|
|
expected = "Ursus americanus American buffalo Bison bison American crow \nCorvus brachyrhynchos American marten Martes americana American racer \nColuber constrictor American woodcock Scolopax minor Anaconda (unidentified)\nEunectes sp. Andean goose Chloephaga melanoptera Ant \nAnteater, australian spiny Tachyglossus aculeatus Anteater, giant Myrmecophaga tridactyla\n\nUrsus americanus Corvus brachyrhynchos Coluber constrictor Eunectes sp. \nAnteater, australian spiny American buffalo American marten American woodcock \nAndean goose Tachyglossus aculeatus Bison bison Martes americana \nScolopax minor Chloephaga melanoptera Anteater, giant American crow \nAmerican racer Anaconda (unidentified) Ant Myrmecophaga tridactyla\n\nEunectes sp. Coluber constrictor Corvus brachyrhynchos Ursus americanus \nAmerican woodcock American marten American buffalo Anteater, australian spiny\nMartes americana Bison bison Tachyglossus aculeatus Andean goose \nAmerican crow Anteater, giant Chloephaga melanoptera Scolopax minor \nMyrmecophaga tridactyla Ant Anaconda (unidentified) American racer \n\nMartes americana American crow Ursus americanus \nAnt Eunectes sp. American woodcock \nCorvus brachyrhynchos American buffalo Anteater, giant \nAndean goose Scolopax minor American racer \nBison bison Myrmecophaga tridactyla Anteater, australian spiny \nAnaconda (unidentified) Coluber constrictor American marten \n Tachyglossus aculeatus Chloephaga melanoptera \n\n"
|
||
|
|
assert render() == expected
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
result = render()
|
||
|
|
print(result)
|
||
|
|
print(repr(result))
|