refactor: improve compactness (#841)

openforis · Aug 6, 2023 · d742498 · d742498
2 parents aa46f65 + 1895740
commit d742498
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/sepal_ui/scripts/utils.py b/sepal_ui/scripts/utils.py
@@ -113,12 +113,8 @@ def get_file_size(filename: Union[str, Path]) -> str:
     """
     file_size = Path(filename).stat().st_size
 
-    if file_size == 0:
-        return "0B"
-
     size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
-
-    i = int(math.floor(math.log(file_size, 1024)))
+    i = int(math.floor(math.log(file_size, 1024))) if file_size > 0 else 0
     s = file_size / (1024**i)
 
     return "{:.1f} {}".format(s, size_name[i])

diff --git a/tests/test_scripts/test_utils.py b/tests/test_scripts/test_utils.py
@@ -92,7 +92,7 @@ def test_get_file_size() -> None:
         stat.return_value.st_size = 0
 
         txt = su.get_file_size("random")
-        assert txt == "0B"
+        assert txt == "0.0 B"
 
     # mock every pow of 1024 to YB
     for i in range(9):