import unittest

from test import test_support

from weakref import proxy

import operator

import copy

import pickle

import os

from random import randrange, shuffle

import sys

class PassThru(Exception):

    pass

def check_pass_thru():

    raise PassThru

    yield 1

class BadCmp:

    def __hash__(self):

        return 1

    def __cmp__(self, other):

        raise RuntimeError

class ReprWrapper:

    'Used to test self-referential repr() calls'

    def __repr__(self):

        return repr(self.value)

class HashCountingInt(int):

    'int-like object that counts the number of times __hash__ is called'

    def __init__(self, *args):

        self.hash_count = 0

    def __hash__(self):

        self.hash_count += 1

        return int.__hash__(self)

class TestJointOps(unittest.TestCase):

    # Tests common to both set and frozenset

    def setUp(self):

        self.word = word = 'simsalabim'

        self.otherword = 'madagascar'

        self.letters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'

        self.s = self.thetype(word)

        self.d = dict.fromkeys(word)

    def test_new_or_init(self):

        self.assertRaises(TypeError, self.thetype, [], 2)

    def test_uniquification(self):

        actual = sorted(self.s)

        expected = sorted(self.d)

        self.assertEqual(actual, expected)

        self.assertRaises(PassThru, self.thetype, check_pass_thru())

        self.assertRaises(TypeError, self.thetype, [[]])

    def test_len(self):

        self.assertEqual(len(self.s), len(self.d))

    def test_contains(self):

        for c in self.letters:

            self.assertEqual(c in self.s, c in self.d)

        self.assertRaises(TypeError, self.s.__contains__, [[]])

        s = self.thetype([frozenset(self.letters)])

        self.assert_(self.thetype(self.letters) in s)

    def test_union(self):

        u = self.s.union(self.otherword)

        for c in self.letters:

            self.assertEqual(c in u, c in self.d or c in self.otherword)

        self.assertEqual(self.s, self.thetype(self.word))

        self.assertEqual(type(u), self.thetype)

        self.assertRaises(PassThru, self.s.union, check_pass_thru())

        self.assertRaises(TypeError, self.s.union, [[]])

        for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

            self.assertEqual(self.thetype('abcba').union(C('cdc')), set('abcd'))

            self.assertEqual(self.thetype('abcba').union(C('efgfe')), set('abcefg'))

            self.assertEqual(self.thetype('abcba').union(C('ccb')), set('abc'))

            self.assertEqual(self.thetype('abcba').union(C('ef')), set('abcef'))

    def test_or(self):

        i = self.s.union(self.otherword)

        self.assertEqual(self.s | set(self.otherword), i)

        self.assertEqual(self.s | frozenset(self.otherword), i)

        try:

            self.s | self.otherword

        except TypeError:

            pass

        else:

            self.fail("s|t did not screen-out general iterables")

    def test_intersection(self):

        i = self.s.intersection(self.otherword)

        for c in self.letters:

            self.assertEqual(c in i, c in self.d and c in self.otherword)

        self.assertEqual(self.s, self.thetype(self.word))

        self.assertEqual(type(i), self.thetype)

        self.assertRaises(PassThru, self.s.intersection, check_pass_thru())

        for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

            self.assertEqual(self.thetype('abcba').intersection(C('cdc')), set('cc'))

            self.assertEqual(self.thetype('abcba').intersection(C('efgfe')), set(''))

            self.assertEqual(self.thetype('abcba').intersection(C('ccb')), set('bc'))

            self.assertEqual(self.thetype('abcba').intersection(C('ef')), set(''))

    def test_and(self):

        i = self.s.intersection(self.otherword)

        self.assertEqual(self.s & set(self.otherword), i)

        self.assertEqual(self.s & frozenset(self.otherword), i)

        try:

            self.s & self.otherword

        except TypeError:

            pass

        else:

            self.fail("s&t did not screen-out general iterables")

    def test_difference(self):

        i = self.s.difference(self.otherword)

        for c in self.letters:

            self.assertEqual(c in i, c in self.d and c not in self.otherword)

        self.assertEqual(self.s, self.thetype(self.word))

        self.assertEqual(type(i), self.thetype)

        self.assertRaises(PassThru, self.s.difference, check_pass_thru())

        self.assertRaises(TypeError, self.s.difference, [[]])

        for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

            self.assertEqual(self.thetype('abcba').difference(C('cdc')), set('ab'))

            self.assertEqual(self.thetype('abcba').difference(C('efgfe')), set('abc'))

            self.assertEqual(self.thetype('abcba').difference(C('ccb')), set('a'))

            self.assertEqual(self.thetype('abcba').difference(C('ef')), set('abc'))

    def test_sub(self):

        i = self.s.difference(self.otherword)

        self.assertEqual(self.s - set(self.otherword), i)

        self.assertEqual(self.s - frozenset(self.otherword), i)

        try:

            self.s - self.otherword

        except TypeError:

            pass

        else:

            self.fail("s-t did not screen-out general iterables")

    def test_symmetric_difference(self):

        i = self.s.symmetric_difference(self.otherword)

        for c in self.letters:

            self.assertEqual(c in i, (c in self.d) ^ (c in self.otherword))

        self.assertEqual(self.s, self.thetype(self.word))

        self.assertEqual(type(i), self.thetype)

        self.assertRaises(PassThru, self.s.symmetric_difference, check_pass_thru())

        self.assertRaises(TypeError, self.s.symmetric_difference, [[]])

        for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

            self.assertEqual(self.thetype('abcba').symmetric_difference(C('cdc')), set('abd'))

            self.assertEqual(self.thetype('abcba').symmetric_difference(C('efgfe')), set('abcefg'))

            self.assertEqual(self.thetype('abcba').symmetric_difference(C('ccb')), set('a'))

            self.assertEqual(self.thetype('abcba').symmetric_difference(C('ef')), set('abcef'))

    def test_xor(self):

        i = self.s.symmetric_difference(self.otherword)

        self.assertEqual(self.s ^ set(self.otherword), i)

        self.assertEqual(self.s ^ frozenset(self.otherword), i)

        try:

            self.s ^ self.otherword

        except TypeError:

            pass

        else:

            self.fail("s^t did not screen-out general iterables")

    def test_equality(self):

        self.assertEqual(self.s, set(self.word))

        self.assertEqual(self.s, frozenset(self.word))

        self.assertEqual(self.s == self.word, False)

        self.assertNotEqual(self.s, set(self.otherword))

        self.assertNotEqual(self.s, frozenset(self.otherword))

        self.assertEqual(self.s != self.word, True)

    def test_setOfFrozensets(self):

        t = map(frozenset, ['abcdef', 'bcd', 'bdcb', 'fed', 'fedccba'])

        s = self.thetype(t)

        self.assertEqual(len(s), 3)

    def test_compare(self):

        self.assertRaises(TypeError, self.s.__cmp__, self.s)

    def test_sub_and_super(self):

        p, q, r = map(self.thetype, ['ab', 'abcde', 'def'])

        self.assert_(p < q)

        self.assert_(p <= q)

        self.assert_(q <= q)

        self.assert_(q > p)

        self.assert_(q >= p)

        self.failIf(q < r)

        self.failIf(q <= r)

        self.failIf(q > r)

        self.failIf(q >= r)

        self.assert_(set('a').issubset('abc'))

        self.assert_(set('abc').issuperset('a'))

        self.failIf(set('a').issubset('cbs'))

        self.failIf(set('cbs').issuperset('a'))

    def test_pickling(self):

        for i in (0, 1, 2):

            p = pickle.dumps(self.s, i)

            dup = pickle.loads(p)

            self.assertEqual(self.s, dup, "%s != %s" % (self.s, dup))

            if type(self.s) not in (set, frozenset):

                self.s.x = 10

                p = pickle.dumps(self.s)

                dup = pickle.loads(p)

                self.assertEqual(self.s.x, dup.x)

    def test_deepcopy(self):

        class Tracer:

            def __init__(self, value):

                self.value = value

            def __hash__(self):

                return self.value

            def __deepcopy__(self, memo=None):

                return Tracer(self.value + 1)

        t = Tracer(10)

        s = self.thetype([t])

        dup = copy.deepcopy(s)

        self.assertNotEqual(id(s), id(dup))

        for elem in dup:

            newt = elem

        self.assertNotEqual(id(t), id(newt))

        self.assertEqual(t.value + 1, newt.value)

    def test_gc(self):

        # Create a nest of cycles to exercise overall ref count check

        class A:

            pass

        s = set(A() for i in xrange(1000))

        for elem in s:

            elem.cycle = s

            elem.sub = elem

            elem.set = set([elem])

    def test_subclass_with_custom_hash(self):

        # Bug #1257731

        class H(self.thetype):

            def __hash__(self):

                return int(id(self) & 0x7fffffff)

        s=H()

        f=set()

        f.add(s)

        self.assert_(s in f)

        f.remove(s)

        f.add(s)

        f.discard(s)

    def test_badcmp(self):

        s = self.thetype([BadCmp()])

        # Detect comparison errors during insertion and lookup

        self.assertRaises(RuntimeError, self.thetype, [BadCmp(), BadCmp()])

        self.assertRaises(RuntimeError, s.__contains__, BadCmp())

        # Detect errors during mutating operations

        if hasattr(s, 'add'):

            self.assertRaises(RuntimeError, s.add, BadCmp())

            self.assertRaises(RuntimeError, s.discard, BadCmp())

            self.assertRaises(RuntimeError, s.remove, BadCmp())

    def test_cyclical_repr(self):

        w = ReprWrapper()

        s = self.thetype([w])

        w.value = s

        name = repr(s).partition('(')[0]    # strip class name from repr string

        self.assertEqual(repr(s), '%s([%s(...)])' % (name, name))

    def test_cyclical_print(self):

        w = ReprWrapper()

        s = self.thetype([w])

        w.value = s

        try:

            fo = open(test_support.TESTFN, "wb")

            print >> fo, s,

            fo.close()

            fo = open(test_support.TESTFN, "rb")

            self.assertEqual(fo.read(), repr(s))

        finally:

            fo.close()

            os.remove(test_support.TESTFN)

    def test_do_not_rehash_dict_keys(self):

        n = 10

        d = dict.fromkeys(map(HashCountingInt, xrange(n)))

        self.assertEqual(sum(elem.hash_count for elem in d), n)

        s = self.thetype(d)

        self.assertEqual(sum(elem.hash_count for elem in d), n)

        s.difference(d)

        self.assertEqual(sum(elem.hash_count for elem in d), n)

        if hasattr(s, 'symmetric_difference_update'):

            s.symmetric_difference_update(d)

        self.assertEqual(sum(elem.hash_count for elem in d), n)

        d2 = dict.fromkeys(set(d))

        self.assertEqual(sum(elem.hash_count for elem in d), n)

        d3 = dict.fromkeys(frozenset(d))

        self.assertEqual(sum(elem.hash_count for elem in d), n)

        d3 = dict.fromkeys(frozenset(d), 123)

        self.assertEqual(sum(elem.hash_count for elem in d), n)

        self.assertEqual(d3, dict.fromkeys(d, 123))

class TestSet(TestJointOps):

    thetype = set

    def test_init(self):

        s = self.thetype()

        s.__init__(self.word)

        self.assertEqual(s, set(self.word))

        s.__init__(self.otherword)

        self.assertEqual(s, set(self.otherword))

        self.assertRaises(TypeError, s.__init__, s, 2);

        self.assertRaises(TypeError, s.__init__, 1);

    def test_constructor_identity(self):

        s = self.thetype(range(3))

        t = self.thetype(s)

        self.assertNotEqual(id(s), id(t))

    def test_hash(self):

        self.assertRaises(TypeError, hash, self.s)

    def test_clear(self):

        self.s.clear()

        self.assertEqual(self.s, set())

        self.assertEqual(len(self.s), 0)

    def test_copy(self):

        dup = self.s.copy()

        self.assertEqual(self.s, dup)

        self.assertNotEqual(id(self.s), id(dup))

    def test_add(self):

        self.s.add('Q')

        self.assert_('Q' in self.s)

        dup = self.s.copy()

        self.s.add('Q')

        self.assertEqual(self.s, dup)

        self.assertRaises(TypeError, self.s.add, [])

    def test_remove(self):

        self.s.remove('a')

        self.assert_('a' not in self.s)

        self.assertRaises(KeyError, self.s.remove, 'Q')

        self.assertRaises(TypeError, self.s.remove, [])

        s = self.thetype([frozenset(self.word)])

        self.assert_(self.thetype(self.word) in s)

        s.remove(self.thetype(self.word))

        self.assert_(self.thetype(self.word) not in s)

        self.assertRaises(KeyError, self.s.remove, self.thetype(self.word))

    def test_remove_keyerror_unpacking(self):

        # bug:  www.python.org/sf/1576657

        for v1 in ['Q', (1,)]:

            try:

                self.s.remove(v1)

            except KeyError, e:

                v2 = e.args[0]

                self.assertEqual(v1, v2)

            else:

                self.fail()

    def test_discard(self):

        self.s.discard('a')

        self.assert_('a' not in self.s)

        self.s.discard('Q')

        self.assertRaises(TypeError, self.s.discard, [])

        s = self.thetype([frozenset(self.word)])

        self.assert_(self.thetype(self.word) in s)

        s.discard(self.thetype(self.word))

        self.assert_(self.thetype(self.word) not in s)

        s.discard(self.thetype(self.word))

    def test_pop(self):

        for i in xrange(len(self.s)):

            elem = self.s.pop()

            self.assert_(elem not in self.s)

        self.assertRaises(KeyError, self.s.pop)

    def test_update(self):

        retval = self.s.update(self.otherword)

        self.assertEqual(retval, None)

        for c in (self.word + self.otherword):

            self.assert_(c in self.s)

        self.assertRaises(PassThru, self.s.update, check_pass_thru())

        self.assertRaises(TypeError, self.s.update, [[]])

        for p, q in (('cdc', 'abcd'), ('efgfe', 'abcefg'), ('ccb', 'abc'), ('ef', 'abcef')):

            for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

                s = self.thetype('abcba')

                self.assertEqual(s.update(C(p)), None)

                self.assertEqual(s, set(q))

    def test_ior(self):

        self.s |= set(self.otherword)

        for c in (self.word + self.otherword):

            self.assert_(c in self.s)

    def test_intersection_update(self):

        retval = self.s.intersection_update(self.otherword)

        self.assertEqual(retval, None)

        for c in (self.word + self.otherword):

            if c in self.otherword and c in self.word:

                self.assert_(c in self.s)

            else:

                self.assert_(c not in self.s)

        self.assertRaises(PassThru, self.s.intersection_update, check_pass_thru())

        self.assertRaises(TypeError, self.s.intersection_update, [[]])

        for p, q in (('cdc', 'c'), ('efgfe', ''), ('ccb', 'bc'), ('ef', '')):

            for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

                s = self.thetype('abcba')

                self.assertEqual(s.intersection_update(C(p)), None)

                self.assertEqual(s, set(q))

    def test_iand(self):

        self.s &= set(self.otherword)

        for c in (self.word + self.otherword):

            if c in self.otherword and c in self.word:

                self.assert_(c in self.s)

            else:

                self.assert_(c not in self.s)

    def test_difference_update(self):

        retval = self.s.difference_update(self.otherword)

        self.assertEqual(retval, None)

        for c in (self.word + self.otherword):

            if c in self.word and c not in self.otherword:

                self.assert_(c in self.s)

            else:

                self.assert_(c not in self.s)

        self.assertRaises(PassThru, self.s.difference_update, check_pass_thru())

        self.assertRaises(TypeError, self.s.difference_update, [[]])

        self.assertRaises(TypeError, self.s.symmetric_difference_update, [[]])

        for p, q in (('cdc', 'ab'), ('efgfe', 'abc'), ('ccb', 'a'), ('ef', 'abc')):

            for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

                s = self.thetype('abcba')

                self.assertEqual(s.difference_update(C(p)), None)

                self.assertEqual(s, set(q))

    def test_isub(self):

        self.s -= set(self.otherword)

        for c in (self.word + self.otherword):

            if c in self.word and c not in self.otherword:

                self.assert_(c in self.s)

            else:

                self.assert_(c not in self.s)

    def test_symmetric_difference_update(self):

        retval = self.s.symmetric_difference_update(self.otherword)

        self.assertEqual(retval, None)

        for c in (self.word + self.otherword):

            if (c in self.word) ^ (c in self.otherword):

                self.assert_(c in self.s)

            else:

                self.assert_(c not in self.s)

        self.assertRaises(PassThru, self.s.symmetric_difference_update, check_pass_thru())

        self.assertRaises(TypeError, self.s.symmetric_difference_update, [[]])

        for p, q in (('cdc', 'abd'), ('efgfe', 'abcefg'), ('ccb', 'a'), ('ef', 'abcef')):

            for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple:

                s = self.thetype('abcba')

                self.assertEqual(s.symmetric_difference_update(C(p)), None)

                self.assertEqual(s, set(q))

    def test_ixor(self):

        self.s ^= set(self.otherword)

        for c in (self.word + self.otherword):

            if (c in self.word) ^ (c in self.otherword):

                self.assert_(c in self.s)

            else:

                self.assert_(c not in self.s)

    def test_inplace_on_self(self):

        t = self.s.copy()

        t |= t

        self.assertEqual(t, self.s)

        t &= t

        self.assertEqual(t, self.s)

        t -= t

        self.assertEqual(t, self.thetype())

        t = self.s.copy()

        t ^= t

        self.assertEqual(t, self.thetype())

    def test_weakref(self):

        s = self.thetype('gallahad')

        p = proxy(s)

        self.assertEqual(str(p), str(s))

        s = None

        self.assertRaises(ReferenceError, str, p)

    # C API test only available in a debug build

    if hasattr(set, "test_c_api"):

        def test_c_api(self):

            self.assertEqual(set('abc').test_c_api(), True)

class SetSubclass(set):

    pass

class TestSetSubclass(TestSet):

    thetype = SetSubclass

class SetSubclassWithKeywordArgs(set):

    def __init__(self, iterable=[], newarg=None):

        set.__init__(self, iterable)

class TestSetSubclassWithKeywordArgs(TestSet):

    def test_keywords_in_subclass(self):

        'SF bug #1486663 -- this used to erroneously raise a TypeError'

        SetSubclassWithKeywordArgs(newarg=1)

class TestFrozenSet(TestJointOps):

    thetype = frozenset

    def test_init(self):

        s = self.thetype(self.word)

        s.__init__(self.otherword)

        self.assertEqual(s, set(self.word))

    def test_singleton_empty_frozenset(self):

        f = frozenset()

        efs = [frozenset(), frozenset([]), frozenset(()), frozenset(''),

               frozenset(), frozenset([]), frozenset(()), frozenset(''),

               frozenset(xrange(0)), frozenset(frozenset()),

               frozenset(f), f]

        # All of the empty frozensets should have just one id()

        self.assertEqual(len(set(map(id, efs))), 1)

    def test_constructor_identity(self):

        s = self.thetype(range(3))

        t = self.thetype(s)

        self.assertEqual(id(s), id(t))