Elep

# testing/pickleable.py
# Copyright (C) 2005-2021 the SQLAlchemy authors and contributors
# <see AUTHORS file>
#
# This module is part of SQLAlchemy and is released under
# the MIT License: http://www.opensource.org/licenses/mit-license.php

"""Classes used in pickling tests, need to be at the module level for
unpickling.
"""

from . import fixtures

class User(fixtures.ComparableEntity):
    pass

class Order(fixtures.ComparableEntity):
    pass

class Dingaling(fixtures.ComparableEntity):
    pass

class EmailUser(User):
    pass

class Address(fixtures.ComparableEntity):
    pass

# TODO: these are kind of arbitrary....
class Child1(fixtures.ComparableEntity):
    pass

class Child2(fixtures.ComparableEntity):
    pass

class Parent(fixtures.ComparableEntity):
    pass

class Screen(object):
    def __init__(self, obj, parent=None):
        self.obj = obj
        self.parent = parent

class Foo(object):
    def __init__(self, moredata):
        self.data = "im data"
        self.stuff = "im stuff"
        self.moredata = moredata

__hash__ = object.__hash__

def __eq__(self, other):
        return (
            other.data == self.data
            and other.stuff == self.stuff
            and other.moredata == self.moredata
        )

class Bar(object):
    def __init__(self, x, y):
        self.x = x
        self.y = y

__hash__ = object.__hash__

def __eq__(self, other):
        return (
            other.__class__ is self.__class__
            and other.x == self.x
            and other.y == self.y
        )

def __str__(self):
        return "Bar(%d, %d)" % (self.x, self.y)

class OldSchool:
    def __init__(self, x, y):
        self.x = x
        self.y = y

def __eq__(self, other):
        return (
            other.__class__ is self.__class__
            and other.x == self.x
            and other.y == self.y
        )

class OldSchoolWithoutCompare:
    def __init__(self, x, y):
        self.x = x
        self.y = y

class BarWithoutCompare(object):
    def __init__(self, x, y):
        self.x = x
        self.y = y

def __str__(self):
        return "Bar(%d, %d)" % (self.x, self.y)

class NotComparable(object):
    def __init__(self, data):
        self.data = data

def __hash__(self):
        return id(self)

def __eq__(self, other):
        return NotImplemented

def __ne__(self, other):
        return NotImplemented

class BrokenComparable(object):
    def __init__(self, data):
        self.data = data

def __hash__(self):
        return id(self)

def __eq__(self, other):
        raise NotImplementedError

def __ne__(self, other):
        raise NotImplementedError