"""Test cli."""

from click.testing import CliRunner
from dragonfly.cli import viz


def test_viz():
    runner = CliRunner()
    result = runner.invoke(viz)
    assert result.exit_code == 0
    assert result.output.startswith('vi')
    assert result.output.endswith('z!\n')
