Fix --disable-shapely

preprocess_cancellation.py

@@ -23,14 +23,6 @@
     shapely = None
 
 
-def sizeof_fmt(num, suffix="B"):
-    for unit in ["", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi"]:
-        if abs(num) < 1024.0:
-            return f"{num:3.1f}{unit}{suffix}"
-        num /= 1024.0
-    return f"{num:.1f}Yi{suffix}"
-
-
 HEADER_MARKER = "; Pre-Processed for Cancel-Object support\n"
 
 
@@ -496,6 +488,7 @@ def _main():
 
     args = argparser.parse_args()
     if args.disable_shapely:
+        global shapely
         shapely = None
 
     for filename in args.gcode:

pyproject.toml

@@ -2,14 +2,14 @@
 line_length = 120
 
 [tool.pytest.ini_options]
-addopts = '-v --cov'
+addopts = '-v --cov=preprocess_cancellation'
 
 [tool.coverage.run]
-source = "./"
+branch = true
 
 [tool.poetry]
 name = "preprocess_cancellation"
-version = "0.1.1"
+version = "0.1.2"
 description = "GCode processor to add klipper cancel-object markers"
 authors = ["Franklyn Tackitt <[email protected]>"]
 repository = "https://github.com/kageurufu/cancelobject-preprocessor"
@@ -26,7 +26,7 @@ shapely = "*"
 
 [tool.poetry.dev-dependencies]
 black = "*"
-coverage = "*"
+coverage = { version = "*", extras = ["toml"] }
 isort = "*"
 matplotlib = "*"
 pyinstaller = "*"