tests are failing

plotdigitizer/plotdigitizer.py

@@ -8,7 +8,7 @@
 from typing_extensions import Annotated
 
 from plotdigitizer import image
-from plotdigitizer import plot
+from plotdigitizer.plot import plot_traj
 
 # Logger
 from loguru import logger
@@ -40,9 +40,9 @@ def digitize_plot(
             " If not given, you will be asked to click on the figure.",
         ),
     ] = [],
-    plot_file: Annotated[
+    plot: Annotated[
         T.Optional[Path],
-        typer.Option("--plot-file", help="Plot the final result. Requires matplotlib"),
+        typer.Option("--plot", help="Plot the final result. Requires matplotlib"),
     ] = None,
     output: Annotated[
         T.Optional[Path],
@@ -89,8 +89,8 @@ def digitize_plot(
     # compute trajectories
     traj = figure.trajectories()
 
-    if plot_file is not None:
-        plot.plot_traj(traj, figure._last(), plot_file)
+    if plot is not None:
+        plot_traj(traj, figure._last(), plot)
 
     outfile = output or f"{infile}.traj.csv"
     with open(outfile, "w") as f:

tests/test_cli.py

@@ -9,14 +9,14 @@
 HERE = Path(__file__).parent.resolve()
 
 
-def _run_cmdline(infile: Path, points, locations):
+def _run_cmdline(infile: Path, points, locations, extra: str = ""):
     cmd = f"plotdigitizer {str(infile)} "
     pts = " ".join([f"-p {','.join(map(str,pt))}" for pt in points])
     locs = " ".join([f"-l {','.join(map(str,pt))}" for pt in locations])
     cmd += f"{pts} {locs}"
     outfile = infile.with_suffix(".result.png")
     trajfile = infile.with_suffix(".result.csv")
-    cmd += f" --plot {str(outfile)} --output {trajfile}"
+    cmd += f" --plot {str(outfile)} --output {trajfile} {extra}"
     r = subprocess.run(cmd, check=True, shell=True)
     assert r.returncode == 0, f"Failed test {r.returncode}"
     return trajfile
@@ -61,5 +61,6 @@ def test_grid():
         HERE / ".." / "figures" / "graph_with_grid.png",
         [(200, 0), (1000, 0), (0, 50)],
         [(269, 69), (1780, 69), (81, 542)],
+        "--remove-grid"
     )
     _check_csv_file(csvfile)