diff --git a/mlair/helpers/helpers.py b/mlair/helpers/helpers.py index dbfe75cee375e01316b020d52957da0231bd91bf..0b97f826ee34a35dc62313ed9350919a94931e62 100644 --- a/mlair/helpers/helpers.py +++ b/mlair/helpers/helpers.py @@ -151,7 +151,10 @@ def relative_round(x: float, sig: int, ceil=False, floor=False) -> float: def get_order(x: float): """Get order of number (as power of 10)""" - return int(np.floor(np.log10(abs(x)))) + if x == 0: + return -np.inf + else: + return int(np.floor(np.log10(abs(x)))) def remove_items(obj: Union[List, Dict, Tuple], items: Any): diff --git a/test/test_helpers/test_helpers.py b/test/test_helpers/test_helpers.py index bcc255bb561d25a11cb394087065a7c44b9daa54..22eaa102544f93511007204e6633de143c3e022c 100644 --- a/test/test_helpers/test_helpers.py +++ b/test/test_helpers/test_helpers.py @@ -259,7 +259,7 @@ class TestGetOrder: def test_get_order_neg_orders(self): assert get_order(.006) == -3 - assert get_order(0) == 0 + assert np.isinf(get_order(0)) assert get_order(0.00622) == -3 assert get_order(0.00022) == -4