Henrik Lemke
ab8b704c8b
With ECO_NO_X=1 or without X, the terminal tweak's detector recording now plots as text (uniplot) just above the status line, redrawn in place as points come in -- same layout as the Qt plot: detector(s) vs position for one adjustable, detectors over positions vs step for several. New eco.utilities.termplot, the wrapper planned in docs/terminal_plots.md: panel_stack() for stacked plots over a shared x, and LiveRegion / LivePlotter for in-place redraws that coexist with a '\r' status line (prints a fresh copy if other output scrolled in between).
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
"""eco.utilities.termplot: stacked text plots and the in-place live region."""
|
|
|
|
import io
|
|
|
|
import pytest
|
|
|
|
from eco.utilities import termplot
|
|
|
|
|
|
def test_live_region_redraws_in_place_when_nothing_scrolled():
|
|
out = io.StringIO()
|
|
region = termplot.LiveRegion(out)
|
|
region.write("help\n")
|
|
region.write("\x1b[2Kstatus 1\r")
|
|
region.show("a\nb")
|
|
first = out.getvalue()
|
|
# first draw: fresh copy, then the status line restored under it
|
|
assert first.endswith("\x1b[2Ka\n\x1b[2Kb\n\x1b[2Kstatus 1\r")
|
|
region.write("\x1b[2Kstatus 2\r") # status updates with \r only
|
|
region.show("c\nd")
|
|
# nothing scrolled: cursor up 2 lines, rewrite, status line untouched
|
|
assert out.getvalue()[len(first):] == (
|
|
"\x1b[2Kstatus 2\r" + "\r\x1b[2A\x1b[2Kc\n\x1b[2Kd\n"
|
|
)
|
|
|
|
|
|
def test_live_region_prints_a_fresh_copy_after_other_output():
|
|
out = io.StringIO()
|
|
region = termplot.LiveRegion(out)
|
|
region.show("a\nb")
|
|
region.write("enter absolute position\n")
|
|
region.write("\x1b[2Kstatus\r")
|
|
region.show("c\nd")
|
|
assert out.getvalue().endswith("\r\x1b[2K\x1b[2Kc\n\x1b[2Kd\n\x1b[2Kstatus\r")
|
|
region.show("e") # height changed: fresh copy as well
|
|
assert out.getvalue().endswith("\x1b[2Ke\n\x1b[2Kstatus\r")
|
|
|
|
|
|
def test_live_region_installs_as_stdout(monkeypatch):
|
|
import sys
|
|
|
|
out = io.StringIO()
|
|
monkeypatch.setattr(sys, "stdout", out)
|
|
region = termplot.LiveRegion().install()
|
|
assert sys.stdout is region
|
|
print("x")
|
|
region.uninstall()
|
|
assert sys.stdout is out and out.getvalue() == "x\n"
|
|
|
|
|
|
def test_panel_stack_shares_x_and_labels_only_the_bottom():
|
|
pytest.importorskip("uniplot")
|
|
text = termplot.panel_stack(
|
|
[
|
|
{"title": "det", "xs": [0, 1, 2], "ys": [1, 3, 2], "highlight_last": True},
|
|
{"title": "mot", "xs": [0, 1, 2], "ys": [0, 0.5, 1]},
|
|
{"title": "empty", "xs": [], "ys": []},
|
|
],
|
|
width=40,
|
|
height=3,
|
|
)
|
|
lines = text.split("\n")
|
|
assert [l.strip() for l in lines if l.strip() in ("det", "mot", "empty")] == [
|
|
"det",
|
|
"mot",
|
|
"empty",
|
|
]
|
|
frame_ends = [i for i, l in enumerate(lines) if l.startswith("└")]
|
|
assert len(frame_ends) == 3
|
|
# x tick labels once, under the last panel only
|
|
assert frame_ends[-1] == len(lines) - 2 and lines[-1].strip().startswith("0")
|
|
assert all(lines[i + 1].strip() in ("mot", "empty") for i in frame_ends[:-1])
|
|
assert len(lines) == 3 * (1 + 2 + 3) + 1
|