mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 21:42:22 +00:00
869cfed125
* merge branch develop with main * bump desktop version * fix the merge * sync Cargo.lock to desktop 2.0.0 * fix cli conftest.py --------- Co-authored-by: deeleeramone <>
36 lines
876 B
Python
36 lines
876 B
Python
"""Tests for StdioOutput — alias for TsvOutput."""
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
|
|
from openbb_cli.outputs.stdio import StdioOutput
|
|
from openbb_cli.outputs.tsv import TsvOutput
|
|
|
|
|
|
@pytest.fixture()
|
|
def stdio_output():
|
|
return StdioOutput()
|
|
|
|
|
|
def test_stdio_is_tsv_subclass():
|
|
assert issubclass(StdioOutput, TsvOutput)
|
|
|
|
|
|
def test_stdio_export_no_output(stdio_output, capsys):
|
|
stdio_output.display(data="x", export=True)
|
|
assert capsys.readouterr().out == ""
|
|
|
|
|
|
def test_stdio_dataframe(stdio_output, capsys):
|
|
df = pd.DataFrame({"x": [10, 20], "y": [30, 40]})
|
|
stdio_output.display(data=df)
|
|
out = capsys.readouterr().out
|
|
assert "x\ty" in out
|
|
assert "10\t30" in out
|
|
|
|
|
|
def test_stdio_no_ansi_in_output(stdio_output, capsys):
|
|
df = pd.DataFrame({"a": [1]})
|
|
stdio_output.display(data=df)
|
|
assert "\x1b[" not in capsys.readouterr().out
|