return output from blacktex.cli.main

Author: nschloe

src/blacktex/cli.py

@@ -8,14 +8,18 @@ def main(argv=None):
     parser = _get_parser()
     args = parser.parse_args(argv)
 
+    stdout = []
     for fl in args.infiles:
         content = fl.read()
         out = blacktex.clean(content, args.keep_comments, args.keep_dollar_math)
         if args.in_place:
             with open(fl.name, "w") as f:
                 f.write(out)
         else:
-            print(out)
+            stdout.append(out)
+
+    if not args.in_place:
+        return "\n".join(stdout)
 
 
 def _get_parser():

test/test_cli.py

@@ -8,12 +8,9 @@ def test_cli():
     with tempfile.TemporaryDirectory() as tmpdir:
         tmpdir = Path(tmpdir)
         infile = tmpdir / "in.tex"
-        outfile = tmpdir / "out.tex"
         with open(infile, "w") as f:
             f.write("a+b=c")
 
-        blacktex.cli.main([str(infile), str(outfile)])
+        stdout = blacktex.cli.main([str(infile)])
 
-        with open(outfile) as f:
-            line = f.read()
-            assert line == "a+b = c"
+        assert stdout == "a+b = c"