Function assert_slots_equal

dask/tests/test_task_spec.py:251–262

(a, b)

def test_pickle():

    def assert_slots_equal(a, b):
        def get_all_slots(obj):
            slots = set()
            for cls in obj.__class__.mro():
                slots.update(getattr(cls, "__slots__", ()))
            return slots

        all_slots = get_all_slots(a) | get_all_slots(a)
        assert all_slots == get_all_slots(a) == get_all_slots(a)
        assert all(getattr(a, slot) == getattr(b, slot) for slot in all_slots)
        assert not hasattr(a, "__dict__")
        assert not hasattr(b, "__dict__")

    t1 = Task("key-1", func, "a", "b")
    t2 = Task("key-2", func, "c", "d")