Source code for rframe.dispatchers

# type: ignore

import math
from numbers import Number

from collections.abc import Iterable, Mapping
from typing import overload
from plum import Dispatcher

from .types import Interval

dispatch = Dispatcher()


@dispatch
def are_equal(x: str, y: str):
    return x == y


@dispatch
def are_equal(x: Number, y: Number):
    if math.isnan(x) and math.isnan(y):
        return True
    return math.isclose(x, y)


@dispatch
def are_equal(x: Iterable, y: Iterable):
    if len(x) != len(y):
        return False
    return all([are_equal(xi, yi) for xi, yi in zip(x, y)])


@dispatch
def are_equal(x: Mapping, y: Mapping):
    if len(x) != len(y):
        return False
    for k, v in x.items():
        if k not in y or not are_equal(v, y[k]):
            return False
    return True


@dispatch
def are_equal(x: Interval, y: Interval):
    return are_equal(x.left, y.left) and are_equal(x.right, y.right)




@dispatch
def are_equal(x: object, y: object):
    if x is None and y is None:
        return True
    return x == y