123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137 |
- # Copyright (c) "Neo4j"
- # Neo4j Sweden AB [https://neo4j.com]
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # https://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- from typing import (
- Tuple,
- TypeVar,
- )
- __all__ = [
- "nano_add",
- "nano_div",
- "nano_divmod",
- "round_half_to_even",
- "symmetric_divmod",
- ]
- def nano_add(x, y):
- """
- Add with nanosecond precission.
- >>> 0.7 + 0.2
- 0.8999999999999999
- >>> -0.7 + 0.2
- -0.49999999999999994
- >>> nano_add(0.7, 0.2)
- 0.9
- >>> nano_add(-0.7, 0.2)
- -0.5
- :param x:
- :param y:
- :returns:
- """
- return (int(1000000000 * x) + int(1000000000 * y)) / 1000000000
- def nano_div(x, y):
- """
- Div with nanosecond precission.
- >>> 0.7 / 0.2
- 3.4999999999999996
- >>> -0.7 / 0.2
- -3.4999999999999996
- >>> nano_div(0.7, 0.2)
- 3.5
- >>> nano_div(-0.7, 0.2)
- -3.5
- :param x:
- :param y:
- :returns:
- """
- return float(1000000000 * x) / int(1000000000 * y)
- def nano_divmod(x, y):
- """
- Divmod with nanosecond precission.
- >>> divmod(0.7, 0.2)
- (3.0, 0.09999999999999992)
- >>> nano_divmod(0.7, 0.2)
- (3, 0.1)
- :param x:
- :param y:
- :returns:
- """
- number = type(x)
- nx = int(1000000000 * x)
- ny = int(1000000000 * y)
- q, r = divmod(nx, ny)
- return int(q), number(r / 1000000000)
- _TDividend = TypeVar("_TDividend", int, float)
- def symmetric_divmod(
- dividend: _TDividend, divisor: float
- ) -> Tuple[int, _TDividend]:
- number = type(dividend)
- if dividend >= 0:
- quotient, remainder = divmod(dividend, divisor)
- return int(quotient), number(remainder)
- else:
- quotient, remainder = divmod(-dividend, divisor)
- return -int(quotient), -number(remainder)
- def round_half_to_even(n):
- """
- Round x.5 towards the nearest even integer.
- >>> round_half_to_even(3)
- 3
- >>> round_half_to_even(3.2)
- 3
- >>> round_half_to_even(3.5)
- 4
- >>> round_half_to_even(3.7)
- 4
- >>> round_half_to_even(4)
- 4
- >>> round_half_to_even(4.2)
- 4
- >>> round_half_to_even(4.5)
- 4
- >>> round_half_to_even(4.7)
- 5
- :param n:
- :returns:
- """
- ten_n = 10 * n
- if ten_n == int(ten_n) and ten_n % 10 == 5:
- up = int(n + 0.5)
- down = int(n - 0.5)
- return up if up % 2 == 0 else down
- else:
- return int(round(n))
|