def test_show_help(capsys):
with pytest.raises(SystemExit):
main(["flu", "-h"])
result = capsys.readouterr()
stdout = result.out
assert stdout.startswith("usage")
def test_from_file(capsys):
with NamedTemporaryFile("w+") as f:
f.write("hello")
f.read()
f_name = f.name
main(["flu", "-f", f_name, "_.map(str.upper)"])
result = capsys.readouterr()
stdout = result.out.strip("\n")
assert stdout == "HELLO"
def test_cli_walk_dirs(capsys):
main(["flu", "walk_dirs().head(2)"])
result = capsys.readouterr()
stdout = result.out.strip("\n").split("\n")
assert len(stdout) == 2
def test_non_iterable_non_none_pipeline(capsys):
main(["flu", '"hello_world"'])
result = capsys.readouterr()
stdout = result.out.strip("\n")
assert stdout == "hello_world"
def test_pass_on_none_pipeline(capsys):
main(["flu", "None"])
result = capsys.readouterr()
stdout = result.out
assert stdout == ""
def test_basic_pipeline(capsys):
main(["flu", "flu(range(5)).collect()"])
result = capsys.readouterr()
stdout = result.out.replace("\n", "")
assert stdout.startswith("0")
def test_show_version(capsys):
main(["flu", "flu(range(5)).collect()"])
result = capsys.readouterr()
stdout = result.out.replace("\n", "")
assert stdout.startswith("0")
def test_glob_imports(capsys):
main(["flu", "flu(env).count()", "-i", "os:environ:env"])
result = capsys.readouterr()
stdout = result.out
assert stdout
