move tests under tests/

pyinfra-dev · Sep 26, 2024 · f4299e1 · f4299e1
1 parent ab8537c
commit f4299e1
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 29 deletions.
diff --git a/pyinfra/facts/util/units.py b/pyinfra/facts/util/units.py
@@ -28,32 +28,3 @@ def parse_size(size: str | int) -> int:
     if isinstance(size, int):
         return size
     return parse_human_readable_size(size)
-
-
-def test_parse_human_readable_size():
-    example_strings = [
-        "1024b",
-        "10.43 KB",
-        "11 GB",
-        "343.1 MB",
-        "10.43KB",
-        "11GB",
-        "343.1MB",
-        "10.43 kb",
-        "11 gb",
-        "343.1 mb",
-        "10.43kb",
-        "11gb",
-        "343.1mb",
-        "1024Kib",
-        "10.43 KiB",
-        "11 GiB",
-        "343.1 MiB",
-        "10.43KiB",
-        "11GiB",
-        "343.1MiB",
-        "10.43 kib",
-        "11 gib",
-    ]
-    for example_string in example_strings:
-        print(example_string, parse_human_readable_size(example_string))
diff --git a/tests/test_units.py b/tests/test_units.py
@@ -0,0 +1,30 @@
+from pyinfra.facts.util.units import parse_human_readable_size
+
+
+def test_parse_human_readable_size():
+    example_strings = [
+        "1024b",
+        "10.43 KB",
+        "11 GB",
+        "343.1 MB",
+        "10.43KB",
+        "11GB",
+        "343.1MB",
+        "10.43 kb",
+        "11 gb",
+        "343.1 mb",
+        "10.43kb",
+        "11gb",
+        "343.1mb",
+        "1024Kib",
+        "10.43 KiB",
+        "11 GiB",
+        "343.1 MiB",
+        "10.43KiB",
+        "11GiB",
+        "343.1MiB",
+        "10.43 kib",
+        "11 gib",
+    ]
+    for example_string in example_strings:
+        print(example_string, parse_human_readable_size(example_string))