1 files changed, 107 insertions, 0 deletions
diff --git a/lib/sqlalchemy/testing/pickleable.py b/lib/sqlalchemy/testing/pickleable.py
new file mode 100644
index 000000000..f5b8b827c
--- /dev/null
+++ b/lib/sqlalchemy/testing/pickleable.py
@@ -0,0 +1,107 @@
+"""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
+