mirror of
https://github.com/RustPython/RustPython.git
synced 2026-10-28 18:59:04 +08:00
Jeong, YunWon
4fab5c966e
* Correct what the non-unix frame stack is for
The comment said non-unix threading builds have no stop-the-world.
CollectStopTheWorld is gated on `feature = "threading"` alone, and
sys._current_frames stops the world on both paths; the field is the
fallback a reader uses when there is no `top_iframe` to materialize from.
Assisted-by: Claude
* Seek by the whole file position on Windows
`SetFilePointer` answers with the low half of the new position and signals
failure with INVALID_SET_FILE_POINTER, which is also that half of a
position four gigabytes in; telling them apart takes the error code, which
this did not read, so such a seek was reported as an error. Deciding
seekability from it also called such a file unseekable.
`SetFilePointerEx` returns the whole position and a success flag of its
own, which also removes the transmute of the position into halves.
Assisted-by: Claude
* memoryview: tell a value of the wrong kind from one that does not fit
Assigning an item reported every packing failure as a TypeError, so
m[0] = 300 on a 'B' view said the value was the wrong type rather than out
of range. Packing now says which of the two it was, and whether the value's
own code raised, in which case that error is the answer as it is:
m[0] = 300 ValueError: invalid value for format 'B'
m[0] = "x" TypeError: invalid type for format 'B'
m[0] = <__index__ that raises> the raised error
`struct` reports both as `struct.error` and is unchanged; the kind travels
beside the exception for the caller that tells them apart.
Also: None was read as a deletion, so m[0] = None answered "cannot delete
memory" instead of packing it; and deleting through the mapping protocol
never reached the read-only check, which comes first.
Assisted-by: Claude
* Raise MemoryError for allocations sized by Python input
os.read, _RawIOBase.read, int.to_bytes, struct.pack, ctypes array
creation and the _ssl RAND functions sized a Vec from a Python-supplied
length with vec![], which calls handle_alloc_error and, under
panic = "abort", ends the process. They now allocate through
vm.new_zeroed_bytes and raise MemoryError.
itertools.product built its pools without checking that
len(iterables) * repeat is representable; it now raises OverflowError
"repeat argument too large" and reserves the pool and index vectors
fallibly. repeat is read as isize, so a negative one raises ValueError
"repeat argument cannot be negative" instead of the conversion's message.
_ssl.RAND_bytes and RAND_pseudo_bytes read n as i32, matching the int
they are declared with.
Assisted-by: Claude
* Collect with the count in the object header
A collection kept the count it was working with in a table keyed by the
object's address, and the objects it had proved reachable in a second one.
Between them they were hashed once per candidate and twice per edge in the
heap, which is where most of a collection over a live heap went.
The count now lives in `PyInner::gc_refs`, with `GcBits::COLLECTING` saying
it is meaningful, and reachability is `gc_refs == GC_REACHABLE` rather than
membership in a set. Step 5 splits the candidates and clears the bit in one
pass. gcbench, five interleaved pairs, median: a live heap of 423k objects
goes from 0.101s to 0.055s and a dead one from 0.402s to 0.383s.
The bits, generation, owner and count take eight bytes between them. A
64-bit header had those eight as the padding its alignment forces, so it is
unchanged at 48 bytes; a 32-bit header grows from 24 to 28.
Assisted-by: Claude
* Take an explicit thread stack size as a floor in debug builds
A debug build already started Python threads on 8 MiB rather than Rust's
2 MiB default, but an explicit threading.stack_size(N) went through
verbatim. test_threading asks for 256 KiB, and starting a thread on that
walked off the end of the stack: the guard page fault landed in the
prologue of ExecutingFrame::run.
Unoptimized, that prologue reserves 80,848 bytes where the optimized one
reserves 656 -- execute_instruction is #[inline(always)] and LLVM only
colors stack slots from opt-level 1, so the frame is the sum of all 200
instruction arms' temporaries rather than the largest. A Python call costs
88,672 bytes of native stack there, and threading's bootstrap is six
frames deep, so 256 KiB holds less than half of what starting a thread
takes.
The floor reaches thread::Builder only; threading.stack_size() still
answers with what was asked for, and release builds are unchanged.
Assisted-by: Claude
* Check the native stack on every frame entry
The C-stack guard ran on one frame entry in eight. That asks the margin to
cover eight frames rather than one, and it does not: an unoptimized frame
entered through native code takes 88,672 bytes against a debug margin of
262,144. A recursion whose steps re-enter that way -- `__add__` calling
itself, a sort key that sorts -- ran off the end of the stack instead of
raising RecursionError. On a debug build `class Add: __add__ = lambda s, o:
s + o; Add() + 1` segfaulted on the main thread; it now raises, as it does
under CPython and in release builds.
enter_iframe checked and then called enter_iframe_unchecked, which checked
again; it now leaves the check to the one call.
Measured on a call-dominated benchmark, five interleaved pairs: instructions
retired go up 0.17%, about four per call, which is the stack pointer read
and the compare.
Assisted-by: Claude
* Take an iterable's length hint when a list is filled from it
`map_py_iter` read `__length_hint__` only to pass it to `PyIterIter`, and
returned an empty vector when the hint was `isize::MAX` or more. Collecting
through `PyResult` dropped the iterator's lower bound, so nothing reserved
the room the hint asked for.
`list()`, `list.extend()` and `list.__iadd__()` now reserve it and report a
hint they cannot honour as `MemoryError`; a hint that leaves no room for the
elements the list already holds is passed over, as `list_extend()` does.
`tuple()` and the other callers keep filling up without reserving.
`length_hint_opt` errors other than the `TypeError` it already turns into
`None` now reach the caller instead of being dropped.
`__iadd__` and `inplace_concat` went through `extract_cloned`, which reads
`__len__` and not `__length_hint__`; both call `PyList::extend` now, the way
`list_inplace_concat()` calls `list_extend()`.
The tuple, list and dict fast paths of `extract_elements_inner` reserve their
known length, which the `collect()` they used dropped.
Assisted-by: Claude
* Drop map.__length_hint__
`map_methods` has no `__length_hint__`, so `operator.length_hint()` on a map
answers 0, not the length of what it draws from.
The method walked into the length hint of every iterator it holds, and a
chain of maps 10000 long overflowed the native stack answering for the
outermost one. It also took the longest of its iterators, where a map stops
at the shortest.
Assisted-by: Claude
* Ask for a length hint where each caller asks for it
`map_py_iter` asked the iterable it was handed, for every caller, and reported
what asking raised. Only some callers ask it: `list_extend()` and
`_PyBytes_FromIterator()` ask the iterable, `PySequence_Tuple()` asks the
iterator, and the bytearray constructor asks nothing. `tuple()`, `min()`,
`max()`, `collections.deque()` and `f(*x)` raised for an iterable whose
`__len__` or `__length_hint__` does, where they answer.
Which object is asked is now the caller's to say. `sorted()` asks the iterable,
being `PySequence_List()`.
`bytes_from_object()` stood in for `PyBytes_FromObject()`, for
`bytearray_extend()` and for the bytearray constructor, which do not agree on
this: the first two ask, the last does not. It is split, and assigning to a
bytearray slice takes the constructor's side with `PyByteArray_FromObject()`.
`list.extend()` counted what it held before the iterable had been asked, where
`list_extend()` reads `Py_SIZE(self)` after. A `__length_hint__` that adds to
the list made the overflow guard read a count too small and raise `MemoryError`
where nothing is wrong; one that empties it made the guard skip a reservation
that cannot be served.
Assisted-by: Claude
* Settle product's pool count before it reads its arguments
`product_new()` checks `repeat` and works out `npools` before it calls
`PySequence_Tuple()` on any argument, and fills the pools `npools` times.
The pools were filled by repeating the arguments `repeat` times instead, which
walks that many steps even with no arguments to repeat: `product(repeat=2**62)`
counted up to it rather than answering `[()]`. The count was also worked out
after the arguments had been read, so a repeat too large to serve ran their
code first.
Assisted-by: Claude
* Let the bool format answer with the error its value raised
`pack_single()` leaves `'?'` to `PyObject_IsTrue()` and returns what that
raised. Packing classified the error instead, so a `ValueError` from a
`__bool__` came back as "memoryview: invalid value for format '?'".
Assisted-by: Claude
* Release a cell's old value after the lock
`PyCell::set` dropped what it replaced while still holding the mutex guarding
the cell contents. A `__del__` running from that drop and reading the same cell
waited on a lock its own caller held, so `del it` on a closure variable whose
value has such a `__del__` deadlocked.
The replaced value is now released once the guard is gone, as `Py_XSETREF`
stores before it decrefs.
Assisted-by: Claude
* Have a set iterator hold the set it iterates
The iterator kept only a reference to the inner hash table, so in
`it = iter(A(*args))` the `A()` temporary was the last owner and died as the
call returned, before `it` was bound. A `__del__` reading `it` there saw an
unbound name and its error was printed and ignored.
The iterator now holds the set object, as `si_set` does, and releases it once
exhausted. That release happens after the lock is dropped, where
`setiter_iternext()` puts its `Py_DECREF(so)` past `Py_END_CRITICAL_SECTION()`,
so a `__del__` that iterates again does not wait on a lock the call holds.
Assisted-by: Claude
* Release an exhausted iterator's container after the lock
`PositionIterInternal::_next` overwrote its `IterStatus::Active` while the
caller still held the mutex around it. Dropping the container there ran any
`__del__` under that lock, and a `__del__` that iterated the same object again
blocked on it.
`exhaust()` now hands the container back instead of dropping it, and
`locked_step()` releases it after the guard. list, list_reverseiterator, tuple,
str, dict and its views and reverse views, bytes, bytearray, memoryview,
array, deque, and the enumerate and sequence iterators all go through it.
Assisted-by: Claude
* Unskip test_free_after_iterating
Assisted-by: Claude
* Stop listing test_set as an environment polluter
`check_free_after_iterating` no longer leaves an ignored exception behind, and
the job that reruns the listed tests ten times fails once one of them stops
polluting.
Assisted-by: Claude
* Drop winsound imports left unused
`TryFromBorrowedObject`, `exceptions`, and `ToWideString` have no reference in
the module, which fails the Windows clippy line under `-Dwarnings`.
Assisted-by: Claude
* Ask nothing where the caller takes no room
`map_py_iter` asked the iterator for a length hint on behalf of every caller
that does not reserve, and reported what asking raised. Those callers ask
nothing at all: `tuple()`, `f(*x)`, `bytearray(x)`, `min()`, `max()` and
`collections.deque()` answer for an iterator whose `__length_hint__` raises,
where they had been raising it.
The answer was also never spent. It reached `PyIterIter` for a `size_hint()`
the push loop does not read, so the lookup and any call it made were work
thrown away: `tuple()` over a generator drops 22% of its instructions, and 17%
over an iterator with a `__length_hint__` written in Python.
Assisted-by: Claude
* Stop asking an iterator how long it is to walk it
`PyIter::iter` and `PyIter::into_iter` asked for a length hint and reported
what asking raised. Nothing spent the answer: it reached `PyIterIter` for a
`size_hint()` that every caller either loops past or drops, since collecting
into a `Result` reports no lower bound.
23 operations answered for an iterator whose `__length_hint__` raises, where
they had been raising it: `set`, `frozenset` and the nine `set` methods that
take an iterable, `dict.fromkeys` and the dict view operators, `array` and
`array.extend`, `all`, `any`, `sum`, `io.writelines`, `math.fsum`,
`math.prod`, and `csv.writerow` and `writerows`.
Over a generator, `set()` drops 16% of its instructions and `all()` 23%.
`str.join` and `bytes.join` do ask, reaching their elements through
`PySequence_Fast()`, which fills a list from the iterator. They take
`iter_sized()`, which is now the only way to ask. `iter_without_hint` is gone,
its callers being what `iter` already does.
Assisted-by: Claude
* Name the matrix jobs rather than let the matrix name them
A generated job name lists every value in the matrix entry, so `cargo check`
carried the booleans its dependencies and `skip_ssl` keys expand to, and the
snippets job carried its test arguments and timeout. Adding or removing a key
renames the check, which drops it from the required list until that list is
edited to match. Emptying `env_polluting_tests` renamed three checks this way.
`Run rust tests` and `clippy` keep the names they had. `cargo check` drops the
booleans from six of its nine, and the snippets job drops its arguments and
timeout from all three.
Assisted-by: Claude
* Keep a sequence iterator active when an element raises
`PositionIterInternal::_next` exhausted the iterator for any non-`Return`
result, so an error from `__getitem__` ended the walk. `iter_iternext()`
lets go of its sequence for `IndexError` and `StopIteration` alone, and
`PyIterReturn::from_getitem_result` has already turned the first of those
into the second, so only `StopIteration` exhausts now.
Both deque iterators keep exhausting on their own mutation guard, which
`deque_iternext()` does by zeroing the counter before it raises; they
share one step function for it, and the message it raises is lowercased
to match the three other sites in the module.
Assisted-by: Claude
* Name the caller in the bytes conversion TypeError
An object with no iteration protocol reached `PyObject_GetIter`'s "not
iterable" message, and `bytearray.extend()` reported bytes. Each entry
point now checks for the protocol first and names itself:
bytes(object()) cannot convert 'object' object to bytes
bytearray(object()) cannot convert 'object' object to bytearray
bytearray().extend(object()) can't extend bytearray with object
Assisted-by: Claude
* Take a strong count too large to hold as reachable
`start_gc_refs` clipped a count at `GC_REACHABLE - 1`, a number the
per-reference subtraction could still walk down to zero and collect a
live object. It now stores `GC_REACHABLE`, and `subtract_gc_ref` leaves
that value alone.
Assisted-by: Claude
* Raise AssertionError where a snippet asserts False
`assert False` is removed under `-O`, which the snippet suite may run.
Assisted-by: Claude
* Keep raising for a collection that moved under its iterator
A deque, set or dict iterator raised once and then read as spent. The
guard is sticky: `deque_iternext()` looks at the deque's state before the
count it keeps, and `dictiter_iternextkey()` and `setiter_iternext()`
write a size no collection can have, so every later call finds the same
thing and raises again. `dequereviter_next()` is the one exception,
looking at its count first, so it runs out after the first raise.
Both deque iterators now carry `dequeiterobject.counter` rather than
reading a length back from the deque, and the dict and set iterators
compare the size they captured against the collection's own every time
they are asked how much is left, which is what makes that answer nothing
from the moment the collection changes rather than only once the
iterator has raised.
The set's message is capitalized to match `setiter_iternext()`.
Measured against 3.14.6, for each of deque, reversed deque, set, dict
and a reversed dict view: the hint after the change, the error, the hint
after the error, and what a later call answers.
Assisted-by: Claude
* Give apt-get update a deadline to fail on
The step waits on `apt-get update`, which has no deadline of its own, so
a source that takes the connection and then stops answering holds the
job until the workflow's own timeout. The retry that disables the
Microsoft and azure-cli sources runs only when the update exits
non-zero, which a held connection never does; three jobs on this branch
sat on this step for 20 minutes to five and a half hours.
Each attempt is now bounded, and the transports are given a timeout, so
a source that stops answering reaches the retry.
Assisted-by: Claude
2225 lines
74 KiB
Python
Vendored
2225 lines
74 KiB
Python
Vendored
import collections.abc
|
|
import copy
|
|
import gc
|
|
import itertools
|
|
import operator
|
|
import pickle
|
|
import re
|
|
import unittest
|
|
import warnings
|
|
import weakref
|
|
from random import randrange, shuffle
|
|
from test import support
|
|
from test.support import warnings_helper
|
|
|
|
|
|
class PassThru(Exception):
|
|
pass
|
|
|
|
def check_pass_thru():
|
|
raise PassThru
|
|
yield 1
|
|
|
|
class CustomHash:
|
|
def __init__(self, hash):
|
|
self.hash = hash
|
|
def __hash__(self):
|
|
return self.hash
|
|
def __repr__(self):
|
|
return f'<CustomHash {self.hash} at {id(self):#x}>'
|
|
|
|
class BadCmp:
|
|
def __hash__(self):
|
|
return 1
|
|
def __eq__(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:
|
|
# 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)
|
|
self.assertRaises(TypeError, set().__init__, a=1)
|
|
|
|
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.assertIn(self.thetype(self.letters), 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.basetype)
|
|
self.assertRaises(PassThru, self.s.union, check_pass_thru())
|
|
self.assertRaises(TypeError, self.s.union, [[]])
|
|
for C in set, frozenset, dict.fromkeys, str, 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'))
|
|
self.assertEqual(self.thetype('abcba').union(C('ef'), C('fg')), set('abcefg'))
|
|
|
|
# Issue #6573
|
|
x = self.thetype()
|
|
self.assertEqual(x.union(set([1]), x, set([2])), self.thetype([1, 2]))
|
|
|
|
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.basetype)
|
|
self.assertRaises(PassThru, self.s.intersection, check_pass_thru())
|
|
for C in set, frozenset, dict.fromkeys, str, 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(''))
|
|
self.assertEqual(self.thetype('abcba').intersection(C('cbcf'), C('bag')), set('b'))
|
|
s = self.thetype('abcba')
|
|
z = s.intersection()
|
|
if self.thetype == frozenset():
|
|
self.assertEqual(id(s), id(z))
|
|
else:
|
|
self.assertNotEqual(id(s), id(z))
|
|
|
|
def test_isdisjoint(self):
|
|
def f(s1, s2):
|
|
'Pure python equivalent of isdisjoint()'
|
|
return not set(s1).intersection(s2)
|
|
for larg in '', 'a', 'ab', 'abc', 'ababac', 'cdc', 'cc', 'efgfe', 'ccb', 'ef':
|
|
s1 = self.thetype(larg)
|
|
for rarg in '', 'a', 'ab', 'abc', 'ababac', 'cdc', 'cc', 'efgfe', 'ccb', 'ef':
|
|
for C in set, frozenset, dict.fromkeys, str, list, tuple:
|
|
s2 = C(rarg)
|
|
actual = s1.isdisjoint(s2)
|
|
expected = f(s1, s2)
|
|
self.assertEqual(actual, expected)
|
|
self.assertTrue(actual is True or actual is False)
|
|
|
|
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.basetype)
|
|
self.assertRaises(PassThru, self.s.difference, check_pass_thru())
|
|
self.assertRaises(TypeError, self.s.difference, [[]])
|
|
for C in set, frozenset, dict.fromkeys, str, 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'))
|
|
self.assertEqual(self.thetype('abcba').difference(), set('abc'))
|
|
self.assertEqual(self.thetype('abcba').difference(C('a'), C('b')), set('c'))
|
|
|
|
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.basetype)
|
|
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, 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_sub_and_super(self):
|
|
p, q, r = map(self.thetype, ['ab', 'abcde', 'def'])
|
|
self.assertTrue(p < q)
|
|
self.assertTrue(p <= q)
|
|
self.assertTrue(q <= q)
|
|
self.assertTrue(q > p)
|
|
self.assertTrue(q >= p)
|
|
self.assertFalse(q < r)
|
|
self.assertFalse(q <= r)
|
|
self.assertFalse(q > r)
|
|
self.assertFalse(q >= r)
|
|
self.assertTrue(set('a').issubset('abc'))
|
|
self.assertTrue(set('abc').issuperset('a'))
|
|
self.assertFalse(set('a').issubset('cbs'))
|
|
self.assertFalse(set('cbs').issuperset('a'))
|
|
|
|
def test_pickling(self):
|
|
for i in range(pickle.HIGHEST_PROTOCOL + 1):
|
|
if type(self.s) not in (set, frozenset):
|
|
self.s.x = ['x']
|
|
self.s.z = ['z']
|
|
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.assertEqual(self.s.x, dup.x)
|
|
self.assertEqual(self.s.z, dup.z)
|
|
self.assertNotHasAttr(self.s, 'y')
|
|
del self.s.x, self.s.z
|
|
|
|
def test_iterator_pickling(self):
|
|
for proto in range(pickle.HIGHEST_PROTOCOL + 1):
|
|
itorg = iter(self.s)
|
|
data = self.thetype(self.s)
|
|
d = pickle.dumps(itorg, proto)
|
|
it = pickle.loads(d)
|
|
# Set iterators unpickle as list iterators due to the
|
|
# undefined order of set items.
|
|
# self.assertEqual(type(itorg), type(it))
|
|
self.assertIsInstance(it, collections.abc.Iterator)
|
|
self.assertEqual(self.thetype(it), data)
|
|
|
|
it = pickle.loads(d)
|
|
try:
|
|
drop = next(it)
|
|
except StopIteration:
|
|
continue
|
|
d = pickle.dumps(it, proto)
|
|
it = pickle.loads(d)
|
|
self.assertEqual(self.thetype(it), data - self.thetype((drop,)))
|
|
|
|
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 range(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.assertIn(s, 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
|
|
if self.thetype == set:
|
|
self.assertEqual(repr(s), '{set(...)}')
|
|
else:
|
|
name = repr(s).partition('(')[0] # strip class name
|
|
self.assertEqual(repr(s), '%s({%s(...)})' % (name, name))
|
|
|
|
def test_do_not_rehash_dict_keys(self):
|
|
n = 10
|
|
d = dict.fromkeys(map(HashCountingInt, range(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))
|
|
|
|
@unittest.expectedFailure # TODO: RUSTPYTHON
|
|
def test_container_iterator(self):
|
|
# Bug #3680: tp_traverse was not implemented for set iterator object
|
|
class C(object):
|
|
pass
|
|
obj = C()
|
|
ref = weakref.ref(obj)
|
|
container = set([obj, 1])
|
|
obj.x = iter(container)
|
|
del obj, container
|
|
gc.collect()
|
|
self.assertTrue(ref() is None, "Cycle was not collected")
|
|
|
|
def test_free_after_iterating(self):
|
|
support.check_free_after_iterating(self, iter, self.thetype)
|
|
|
|
class TestSet(TestJointOps, unittest.TestCase):
|
|
thetype = set
|
|
basetype = 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_set_literal(self):
|
|
s = set([1,2,3])
|
|
t = {1,2,3}
|
|
self.assertEqual(s, t)
|
|
|
|
def test_set_literal_insertion_order(self):
|
|
# SF Issue #26020 -- Expect left to right insertion
|
|
s = {1, 1.0, True}
|
|
self.assertEqual(len(s), 1)
|
|
stored_value = s.pop()
|
|
self.assertEqual(type(stored_value), int)
|
|
|
|
def test_set_literal_evaluation_order(self):
|
|
# Expect left to right expression evaluation
|
|
events = []
|
|
def record(obj):
|
|
events.append(obj)
|
|
s = {record(1), record(2), record(3)}
|
|
self.assertEqual(events, [1, 2, 3])
|
|
|
|
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))
|
|
self.assertEqual(type(dup), self.basetype)
|
|
|
|
def test_add(self):
|
|
self.s.add('Q')
|
|
self.assertIn('Q', 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.assertNotIn('a', self.s)
|
|
self.assertRaises(KeyError, self.s.remove, 'Q')
|
|
self.assertRaises(TypeError, self.s.remove, [])
|
|
s = self.thetype([frozenset(self.word)])
|
|
self.assertIn(self.thetype(self.word), s)
|
|
s.remove(self.thetype(self.word))
|
|
self.assertNotIn(self.thetype(self.word), s)
|
|
self.assertRaises(KeyError, self.s.remove, self.thetype(self.word))
|
|
|
|
def test_remove_keyerror_unpacking(self):
|
|
# https://bugs.python.org/issue1576657
|
|
for v1 in ['Q', (1,)]:
|
|
try:
|
|
self.s.remove(v1)
|
|
except KeyError as e:
|
|
v2 = e.args[0]
|
|
self.assertEqual(v1, v2)
|
|
else:
|
|
self.fail()
|
|
|
|
def test_remove_keyerror_set(self):
|
|
key = self.thetype([3, 4])
|
|
try:
|
|
self.s.remove(key)
|
|
except KeyError as e:
|
|
self.assertTrue(e.args[0] is key,
|
|
"KeyError should be {0}, not {1}".format(key,
|
|
e.args[0]))
|
|
else:
|
|
self.fail()
|
|
|
|
def test_discard(self):
|
|
self.s.discard('a')
|
|
self.assertNotIn('a', self.s)
|
|
self.s.discard('Q')
|
|
self.assertRaises(TypeError, self.s.discard, [])
|
|
s = self.thetype([frozenset(self.word)])
|
|
self.assertIn(self.thetype(self.word), s)
|
|
s.discard(self.thetype(self.word))
|
|
self.assertNotIn(self.thetype(self.word), s)
|
|
s.discard(self.thetype(self.word))
|
|
|
|
def test_pop(self):
|
|
for i in range(len(self.s)):
|
|
elem = self.s.pop()
|
|
self.assertNotIn(elem, 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.assertIn(c, 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, list, tuple:
|
|
s = self.thetype('abcba')
|
|
self.assertEqual(s.update(C(p)), None)
|
|
self.assertEqual(s, set(q))
|
|
for p in ('cdc', 'efgfe', 'ccb', 'ef', 'abcda'):
|
|
q = 'ahi'
|
|
for C in set, frozenset, dict.fromkeys, str, list, tuple:
|
|
s = self.thetype('abcba')
|
|
self.assertEqual(s.update(C(p), C(q)), None)
|
|
self.assertEqual(s, set(s) | set(p) | set(q))
|
|
|
|
def test_ior(self):
|
|
self.s |= set(self.otherword)
|
|
for c in (self.word + self.otherword):
|
|
self.assertIn(c, 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.assertIn(c, self.s)
|
|
else:
|
|
self.assertNotIn(c, 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, list, tuple:
|
|
s = self.thetype('abcba')
|
|
self.assertEqual(s.intersection_update(C(p)), None)
|
|
self.assertEqual(s, set(q))
|
|
ss = 'abcba'
|
|
s = self.thetype(ss)
|
|
t = 'cbc'
|
|
self.assertEqual(s.intersection_update(C(p), C(t)), None)
|
|
self.assertEqual(s, set('abcba')&set(p)&set(t))
|
|
|
|
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.assertIn(c, self.s)
|
|
else:
|
|
self.assertNotIn(c, 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.assertIn(c, self.s)
|
|
else:
|
|
self.assertNotIn(c, 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, list, tuple:
|
|
s = self.thetype('abcba')
|
|
self.assertEqual(s.difference_update(C(p)), None)
|
|
self.assertEqual(s, set(q))
|
|
|
|
s = self.thetype('abcdefghih')
|
|
s.difference_update()
|
|
self.assertEqual(s, self.thetype('abcdefghih'))
|
|
|
|
s = self.thetype('abcdefghih')
|
|
s.difference_update(C('aba'))
|
|
self.assertEqual(s, self.thetype('cdefghih'))
|
|
|
|
s = self.thetype('abcdefghih')
|
|
s.difference_update(C('cdc'), C('aba'))
|
|
self.assertEqual(s, self.thetype('efghih'))
|
|
|
|
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.assertIn(c, self.s)
|
|
else:
|
|
self.assertNotIn(c, 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.assertIn(c, self.s)
|
|
else:
|
|
self.assertNotIn(c, 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, 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.assertIn(c, self.s)
|
|
else:
|
|
self.assertNotIn(c, 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 = weakref.proxy(s)
|
|
self.assertEqual(str(p), str(s))
|
|
s = None
|
|
support.gc_collect() # For PyPy or other GCs.
|
|
self.assertRaises(ReferenceError, str, p)
|
|
|
|
def test_rich_compare(self):
|
|
class TestRichSetCompare:
|
|
def __gt__(self, some_set):
|
|
self.gt_called = True
|
|
return False
|
|
def __lt__(self, some_set):
|
|
self.lt_called = True
|
|
return False
|
|
def __ge__(self, some_set):
|
|
self.ge_called = True
|
|
return False
|
|
def __le__(self, some_set):
|
|
self.le_called = True
|
|
return False
|
|
|
|
# This first tries the builtin rich set comparison, which doesn't know
|
|
# how to handle the custom object. Upon returning NotImplemented, the
|
|
# corresponding comparison on the right object is invoked.
|
|
myset = {1, 2, 3}
|
|
|
|
myobj = TestRichSetCompare()
|
|
myset < myobj
|
|
self.assertTrue(myobj.gt_called)
|
|
|
|
myobj = TestRichSetCompare()
|
|
myset > myobj
|
|
self.assertTrue(myobj.lt_called)
|
|
|
|
myobj = TestRichSetCompare()
|
|
myset <= myobj
|
|
self.assertTrue(myobj.ge_called)
|
|
|
|
myobj = TestRichSetCompare()
|
|
myset >= myobj
|
|
self.assertTrue(myobj.le_called)
|
|
|
|
def test_set_membership(self):
|
|
myfrozenset = frozenset(range(3))
|
|
myset = {myfrozenset, "abc", 1}
|
|
self.assertIn(set(range(3)), myset)
|
|
self.assertNotIn(set(range(1)), myset)
|
|
myset.discard(set(range(3)))
|
|
self.assertEqual(myset, {"abc", 1})
|
|
self.assertRaises(KeyError, myset.remove, set(range(1)))
|
|
self.assertRaises(KeyError, myset.remove, set(range(3)))
|
|
|
|
def test_unhashable_element(self):
|
|
myset = {'a'}
|
|
elem = [1, 2, 3]
|
|
|
|
def check_unhashable_element():
|
|
msg = "cannot use 'list' as a set element (unhashable type: 'list')"
|
|
return self.assertRaisesRegex(TypeError, re.escape(msg))
|
|
|
|
with check_unhashable_element():
|
|
elem in myset
|
|
with check_unhashable_element():
|
|
myset.add(elem)
|
|
with check_unhashable_element():
|
|
myset.discard(elem)
|
|
|
|
# Only TypeError exception is overriden,
|
|
# other exceptions are left unchanged.
|
|
class HashError:
|
|
def __hash__(self):
|
|
raise KeyError('error')
|
|
|
|
elem2 = HashError()
|
|
with self.assertRaises(KeyError):
|
|
elem2 in myset
|
|
with self.assertRaises(KeyError):
|
|
myset.add(elem2)
|
|
with self.assertRaises(KeyError):
|
|
myset.discard(elem2)
|
|
|
|
def test_hash_collision_remove_add(self):
|
|
self.maxDiff = None
|
|
# There should be enough space, so all elements with unique hash
|
|
# will be placed in corresponding cells without collision.
|
|
n = 64
|
|
elems = [CustomHash(h) for h in range(n)]
|
|
# Elements with hash collision.
|
|
a = CustomHash(n)
|
|
b = CustomHash(n)
|
|
elems += [a, b]
|
|
s = self.thetype(elems)
|
|
self.assertEqual(len(s), len(elems), s)
|
|
s.remove(a)
|
|
# "a" has been replaced with a dummy.
|
|
del elems[n]
|
|
self.assertEqual(len(s), len(elems), s)
|
|
self.assertEqual(s, set(elems))
|
|
s.add(b)
|
|
# "b" should not replace the dummy.
|
|
self.assertEqual(len(s), len(elems), s)
|
|
self.assertEqual(s, set(elems))
|
|
|
|
|
|
class SetSubclass(set):
|
|
pass
|
|
|
|
class TestSetSubclass(TestSet):
|
|
thetype = SetSubclass
|
|
basetype = set
|
|
|
|
def test_keywords_in_subclass(self):
|
|
class subclass(set):
|
|
pass
|
|
u = subclass([1, 2])
|
|
self.assertIs(type(u), subclass)
|
|
self.assertEqual(set(u), {1, 2})
|
|
with self.assertRaises(TypeError):
|
|
subclass(sequence=())
|
|
|
|
class subclass_with_init(set):
|
|
def __init__(self, arg, newarg=None):
|
|
super().__init__(arg)
|
|
self.newarg = newarg
|
|
u = subclass_with_init([1, 2], newarg=3)
|
|
self.assertIs(type(u), subclass_with_init)
|
|
self.assertEqual(set(u), {1, 2})
|
|
self.assertEqual(u.newarg, 3)
|
|
|
|
class subclass_with_new(set):
|
|
def __new__(cls, arg, newarg=None):
|
|
self = super().__new__(cls, arg)
|
|
self.newarg = newarg
|
|
return self
|
|
u = subclass_with_new([1, 2])
|
|
self.assertIs(type(u), subclass_with_new)
|
|
self.assertEqual(set(u), {1, 2})
|
|
self.assertIsNone(u.newarg)
|
|
# disallow kwargs in __new__ only (https://bugs.python.org/issue43413#msg402000)
|
|
with self.assertRaises(TypeError):
|
|
subclass_with_new([1, 2], newarg=3)
|
|
|
|
|
|
class TestFrozenSet(TestJointOps, unittest.TestCase):
|
|
thetype = frozenset
|
|
basetype = frozenset
|
|
|
|
def test_init(self):
|
|
s = self.thetype(self.word)
|
|
s.__init__(self.otherword)
|
|
self.assertEqual(s, set(self.word))
|
|
|
|
def test_constructor_identity(self):
|
|
s = self.thetype(range(3))
|
|
t = self.thetype(s)
|
|
self.assertEqual(id(s), id(t))
|
|
|
|
def test_hash(self):
|
|
self.assertEqual(hash(self.thetype('abcdeb')),
|
|
hash(self.thetype('ebecda')))
|
|
|
|
# make sure that all permutations give the same hash value
|
|
n = 100
|
|
seq = [randrange(n) for i in range(n)]
|
|
results = set()
|
|
for i in range(200):
|
|
shuffle(seq)
|
|
results.add(hash(self.thetype(seq)))
|
|
self.assertEqual(len(results), 1)
|
|
|
|
def test_copy(self):
|
|
dup = self.s.copy()
|
|
self.assertEqual(id(self.s), id(dup))
|
|
|
|
def test_frozen_as_dictkey(self):
|
|
seq = list(range(10)) + list('abcdefg') + ['apple']
|
|
key1 = self.thetype(seq)
|
|
key2 = self.thetype(reversed(seq))
|
|
self.assertEqual(key1, key2)
|
|
self.assertNotEqual(id(key1), id(key2))
|
|
d = {}
|
|
d[key1] = 42
|
|
self.assertEqual(d[key2], 42)
|
|
|
|
def test_hash_caching(self):
|
|
f = self.thetype('abcdcda')
|
|
self.assertEqual(hash(f), hash(f))
|
|
|
|
def test_hash_effectiveness(self):
|
|
n = 13
|
|
hashvalues = set()
|
|
addhashvalue = hashvalues.add
|
|
elemmasks = [(i+1, 1<<i) for i in range(n)]
|
|
for i in range(2**n):
|
|
addhashvalue(hash(frozenset([e for e, m in elemmasks if m&i])))
|
|
self.assertEqual(len(hashvalues), 2**n)
|
|
|
|
def zf_range(n):
|
|
# https://en.wikipedia.org/wiki/Set-theoretic_definition_of_natural_numbers
|
|
nums = [frozenset()]
|
|
for i in range(n-1):
|
|
num = frozenset(nums)
|
|
nums.append(num)
|
|
return nums[:n]
|
|
|
|
def powerset(s):
|
|
for i in range(len(s)+1):
|
|
yield from map(frozenset, itertools.combinations(s, i))
|
|
|
|
for n in range(18):
|
|
t = 2 ** n
|
|
mask = t - 1
|
|
for nums in (range, zf_range):
|
|
u = len({h & mask for h in map(hash, powerset(nums(n)))})
|
|
self.assertGreater(4*u, t)
|
|
|
|
class FrozenSetSubclass(frozenset):
|
|
pass
|
|
|
|
class TestFrozenSetSubclass(TestFrozenSet):
|
|
thetype = FrozenSetSubclass
|
|
basetype = frozenset
|
|
|
|
def test_keywords_in_subclass(self):
|
|
class subclass(frozenset):
|
|
pass
|
|
u = subclass([1, 2])
|
|
self.assertIs(type(u), subclass)
|
|
self.assertEqual(set(u), {1, 2})
|
|
with self.assertRaises(TypeError):
|
|
subclass(sequence=())
|
|
|
|
class subclass_with_init(frozenset):
|
|
def __init__(self, arg, newarg=None):
|
|
self.newarg = newarg
|
|
u = subclass_with_init([1, 2], newarg=3)
|
|
self.assertIs(type(u), subclass_with_init)
|
|
self.assertEqual(set(u), {1, 2})
|
|
self.assertEqual(u.newarg, 3)
|
|
|
|
class subclass_with_new(frozenset):
|
|
def __new__(cls, arg, newarg=None):
|
|
self = super().__new__(cls, arg)
|
|
self.newarg = newarg
|
|
return self
|
|
u = subclass_with_new([1, 2], newarg=3)
|
|
self.assertIs(type(u), subclass_with_new)
|
|
self.assertEqual(set(u), {1, 2})
|
|
self.assertEqual(u.newarg, 3)
|
|
|
|
def test_constructor_identity(self):
|
|
s = self.thetype(range(3))
|
|
t = self.thetype(s)
|
|
self.assertNotEqual(id(s), id(t))
|
|
|
|
def test_copy(self):
|
|
dup = self.s.copy()
|
|
self.assertNotEqual(id(self.s), id(dup))
|
|
|
|
def test_nested_empty_constructor(self):
|
|
s = self.thetype()
|
|
t = self.thetype(s)
|
|
self.assertEqual(s, t)
|
|
|
|
def test_singleton_empty_frozenset(self):
|
|
Frozenset = self.thetype
|
|
f = frozenset()
|
|
F = Frozenset()
|
|
efs = [Frozenset(), Frozenset([]), Frozenset(()), Frozenset(''),
|
|
Frozenset(), Frozenset([]), Frozenset(()), Frozenset(''),
|
|
Frozenset(range(0)), Frozenset(Frozenset()),
|
|
Frozenset(frozenset()), f, F, Frozenset(f), Frozenset(F)]
|
|
# All empty frozenset subclass instances should have different ids
|
|
self.assertEqual(len(set(map(id, efs))), len(efs))
|
|
|
|
|
|
class SetSubclassWithSlots(set):
|
|
__slots__ = ('x', 'y', '__dict__')
|
|
|
|
class TestSetSubclassWithSlots(unittest.TestCase):
|
|
thetype = SetSubclassWithSlots
|
|
setUp = TestJointOps.setUp
|
|
test_pickling = TestJointOps.test_pickling
|
|
|
|
@unittest.expectedFailure # TODO: RUSTPYTHON
|
|
def test_pickling(self):
|
|
return super().test_pickling()
|
|
|
|
class FrozenSetSubclassWithSlots(frozenset):
|
|
__slots__ = ('x', 'y', '__dict__')
|
|
|
|
class TestFrozenSetSubclassWithSlots(TestSetSubclassWithSlots):
|
|
thetype = FrozenSetSubclassWithSlots
|
|
|
|
# Tests taken from test_sets.py =============================================
|
|
|
|
empty_set = set()
|
|
|
|
#==============================================================================
|
|
|
|
class TestBasicOps:
|
|
|
|
def test_repr(self):
|
|
if self.repr is not None:
|
|
self.assertEqual(repr(self.set), self.repr)
|
|
|
|
def check_repr_against_values(self):
|
|
text = repr(self.set)
|
|
self.assertStartsWith(text, '{')
|
|
self.assertEndsWith(text, '}')
|
|
|
|
result = text[1:-1].split(', ')
|
|
result.sort()
|
|
sorted_repr_values = [repr(value) for value in self.values]
|
|
sorted_repr_values.sort()
|
|
self.assertEqual(result, sorted_repr_values)
|
|
|
|
def test_length(self):
|
|
self.assertEqual(len(self.set), self.length)
|
|
|
|
def test_self_equality(self):
|
|
self.assertEqual(self.set, self.set)
|
|
|
|
def test_equivalent_equality(self):
|
|
self.assertEqual(self.set, self.dup)
|
|
|
|
def test_copy(self):
|
|
self.assertEqual(self.set.copy(), self.dup)
|
|
|
|
def test_self_union(self):
|
|
result = self.set | self.set
|
|
self.assertEqual(result, self.dup)
|
|
|
|
def test_empty_union(self):
|
|
result = self.set | empty_set
|
|
self.assertEqual(result, self.dup)
|
|
|
|
def test_union_empty(self):
|
|
result = empty_set | self.set
|
|
self.assertEqual(result, self.dup)
|
|
|
|
def test_self_intersection(self):
|
|
result = self.set & self.set
|
|
self.assertEqual(result, self.dup)
|
|
|
|
def test_empty_intersection(self):
|
|
result = self.set & empty_set
|
|
self.assertEqual(result, empty_set)
|
|
|
|
def test_intersection_empty(self):
|
|
result = empty_set & self.set
|
|
self.assertEqual(result, empty_set)
|
|
|
|
def test_self_isdisjoint(self):
|
|
result = self.set.isdisjoint(self.set)
|
|
self.assertEqual(result, not self.set)
|
|
|
|
def test_empty_isdisjoint(self):
|
|
result = self.set.isdisjoint(empty_set)
|
|
self.assertEqual(result, True)
|
|
|
|
def test_isdisjoint_empty(self):
|
|
result = empty_set.isdisjoint(self.set)
|
|
self.assertEqual(result, True)
|
|
|
|
def test_self_symmetric_difference(self):
|
|
result = self.set ^ self.set
|
|
self.assertEqual(result, empty_set)
|
|
|
|
def test_empty_symmetric_difference(self):
|
|
result = self.set ^ empty_set
|
|
self.assertEqual(result, self.set)
|
|
|
|
def test_self_difference(self):
|
|
result = self.set - self.set
|
|
self.assertEqual(result, empty_set)
|
|
|
|
def test_empty_difference(self):
|
|
result = self.set - empty_set
|
|
self.assertEqual(result, self.dup)
|
|
|
|
def test_empty_difference_rev(self):
|
|
result = empty_set - self.set
|
|
self.assertEqual(result, empty_set)
|
|
|
|
def test_iteration(self):
|
|
for v in self.set:
|
|
self.assertIn(v, self.values)
|
|
setiter = iter(self.set)
|
|
self.assertEqual(setiter.__length_hint__(), len(self.set))
|
|
|
|
def test_pickling(self):
|
|
for proto in range(pickle.HIGHEST_PROTOCOL + 1):
|
|
p = pickle.dumps(self.set, proto)
|
|
copy = pickle.loads(p)
|
|
self.assertEqual(self.set, copy,
|
|
"%s != %s" % (self.set, copy))
|
|
|
|
def test_issue_37219(self):
|
|
with self.assertRaises(TypeError):
|
|
set().difference(123)
|
|
with self.assertRaises(TypeError):
|
|
set().difference_update(123)
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestBasicOpsEmpty(TestBasicOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.case = "empty set"
|
|
self.values = []
|
|
self.set = set(self.values)
|
|
self.dup = set(self.values)
|
|
self.length = 0
|
|
self.repr = "set()"
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestBasicOpsSingleton(TestBasicOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.case = "unit set (number)"
|
|
self.values = [3]
|
|
self.set = set(self.values)
|
|
self.dup = set(self.values)
|
|
self.length = 1
|
|
self.repr = "{3}"
|
|
|
|
def test_in(self):
|
|
self.assertIn(3, self.set)
|
|
|
|
def test_not_in(self):
|
|
self.assertNotIn(2, self.set)
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestBasicOpsTuple(TestBasicOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.case = "unit set (tuple)"
|
|
self.values = [(0, "zero")]
|
|
self.set = set(self.values)
|
|
self.dup = set(self.values)
|
|
self.length = 1
|
|
self.repr = "{(0, 'zero')}"
|
|
|
|
def test_in(self):
|
|
self.assertIn((0, "zero"), self.set)
|
|
|
|
def test_not_in(self):
|
|
self.assertNotIn(9, self.set)
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestBasicOpsTriple(TestBasicOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.case = "triple set"
|
|
self.values = [0, "zero", operator.add]
|
|
self.set = set(self.values)
|
|
self.dup = set(self.values)
|
|
self.length = 3
|
|
self.repr = None
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestBasicOpsString(TestBasicOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.case = "string set"
|
|
self.values = ["a", "b", "c"]
|
|
self.set = set(self.values)
|
|
self.dup = set(self.values)
|
|
self.length = 3
|
|
|
|
def test_repr(self):
|
|
self.check_repr_against_values()
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestBasicOpsBytes(TestBasicOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.case = "bytes set"
|
|
self.values = [b"a", b"b", b"c"]
|
|
self.set = set(self.values)
|
|
self.dup = set(self.values)
|
|
self.length = 3
|
|
|
|
def test_repr(self):
|
|
self.check_repr_against_values()
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestBasicOpsMixedStringBytes(TestBasicOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.enterContext(warnings_helper.check_warnings())
|
|
warnings.simplefilter('ignore', BytesWarning)
|
|
self.case = "string and bytes set"
|
|
self.values = ["a", "b", b"a", b"b"]
|
|
self.set = set(self.values)
|
|
self.dup = set(self.values)
|
|
self.length = 4
|
|
|
|
def test_repr(self):
|
|
self.check_repr_against_values()
|
|
|
|
#==============================================================================
|
|
|
|
def baditer():
|
|
raise TypeError
|
|
yield True
|
|
|
|
def gooditer():
|
|
yield True
|
|
|
|
class TestExceptionPropagation(unittest.TestCase):
|
|
"""SF 628246: Set constructor should not trap iterator TypeErrors"""
|
|
|
|
def test_instanceWithException(self):
|
|
self.assertRaises(TypeError, set, baditer())
|
|
|
|
def test_instancesWithoutException(self):
|
|
# All of these iterables should load without exception.
|
|
set([1,2,3])
|
|
set((1,2,3))
|
|
set({'one':1, 'two':2, 'three':3})
|
|
set(range(3))
|
|
set('abc')
|
|
set(gooditer())
|
|
|
|
def test_changingSizeWhileIterating(self):
|
|
s = set([1,2,3])
|
|
try:
|
|
for i in s:
|
|
s.update([4])
|
|
except RuntimeError:
|
|
pass
|
|
else:
|
|
self.fail("no exception when changing size during iteration")
|
|
|
|
#==============================================================================
|
|
|
|
class TestSetOfSets(unittest.TestCase):
|
|
def test_constructor(self):
|
|
inner = frozenset([1])
|
|
outer = set([inner])
|
|
element = outer.pop()
|
|
self.assertEqual(type(element), frozenset)
|
|
outer.add(inner) # Rebuild set of sets with .add method
|
|
outer.remove(inner)
|
|
self.assertEqual(outer, set()) # Verify that remove worked
|
|
outer.discard(inner) # Absence of KeyError indicates working fine
|
|
|
|
#==============================================================================
|
|
|
|
class TestBinaryOps(unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set((2, 4, 6))
|
|
|
|
def test_eq(self): # SF bug 643115
|
|
self.assertEqual(self.set, set({2:1,4:3,6:5}))
|
|
|
|
def test_union_subset(self):
|
|
result = self.set | set([2])
|
|
self.assertEqual(result, set((2, 4, 6)))
|
|
|
|
def test_union_superset(self):
|
|
result = self.set | set([2, 4, 6, 8])
|
|
self.assertEqual(result, set([2, 4, 6, 8]))
|
|
|
|
def test_union_overlap(self):
|
|
result = self.set | set([3, 4, 5])
|
|
self.assertEqual(result, set([2, 3, 4, 5, 6]))
|
|
|
|
def test_union_non_overlap(self):
|
|
result = self.set | set([8])
|
|
self.assertEqual(result, set([2, 4, 6, 8]))
|
|
|
|
def test_intersection_subset(self):
|
|
result = self.set & set((2, 4))
|
|
self.assertEqual(result, set((2, 4)))
|
|
|
|
def test_intersection_superset(self):
|
|
result = self.set & set([2, 4, 6, 8])
|
|
self.assertEqual(result, set([2, 4, 6]))
|
|
|
|
def test_intersection_overlap(self):
|
|
result = self.set & set([3, 4, 5])
|
|
self.assertEqual(result, set([4]))
|
|
|
|
def test_intersection_non_overlap(self):
|
|
result = self.set & set([8])
|
|
self.assertEqual(result, empty_set)
|
|
|
|
def test_isdisjoint_subset(self):
|
|
result = self.set.isdisjoint(set((2, 4)))
|
|
self.assertEqual(result, False)
|
|
|
|
def test_isdisjoint_superset(self):
|
|
result = self.set.isdisjoint(set([2, 4, 6, 8]))
|
|
self.assertEqual(result, False)
|
|
|
|
def test_isdisjoint_overlap(self):
|
|
result = self.set.isdisjoint(set([3, 4, 5]))
|
|
self.assertEqual(result, False)
|
|
|
|
def test_isdisjoint_non_overlap(self):
|
|
result = self.set.isdisjoint(set([8]))
|
|
self.assertEqual(result, True)
|
|
|
|
def test_sym_difference_subset(self):
|
|
result = self.set ^ set((2, 4))
|
|
self.assertEqual(result, set([6]))
|
|
|
|
def test_sym_difference_superset(self):
|
|
result = self.set ^ set((2, 4, 6, 8))
|
|
self.assertEqual(result, set([8]))
|
|
|
|
def test_sym_difference_overlap(self):
|
|
result = self.set ^ set((3, 4, 5))
|
|
self.assertEqual(result, set([2, 3, 5, 6]))
|
|
|
|
def test_sym_difference_non_overlap(self):
|
|
result = self.set ^ set([8])
|
|
self.assertEqual(result, set([2, 4, 6, 8]))
|
|
|
|
#==============================================================================
|
|
|
|
class TestUpdateOps(unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set((2, 4, 6))
|
|
|
|
def test_union_subset(self):
|
|
self.set |= set([2])
|
|
self.assertEqual(self.set, set((2, 4, 6)))
|
|
|
|
def test_union_superset(self):
|
|
self.set |= set([2, 4, 6, 8])
|
|
self.assertEqual(self.set, set([2, 4, 6, 8]))
|
|
|
|
def test_union_overlap(self):
|
|
self.set |= set([3, 4, 5])
|
|
self.assertEqual(self.set, set([2, 3, 4, 5, 6]))
|
|
|
|
def test_union_non_overlap(self):
|
|
self.set |= set([8])
|
|
self.assertEqual(self.set, set([2, 4, 6, 8]))
|
|
|
|
def test_union_method_call(self):
|
|
self.set.update(set([3, 4, 5]))
|
|
self.assertEqual(self.set, set([2, 3, 4, 5, 6]))
|
|
|
|
def test_intersection_subset(self):
|
|
self.set &= set((2, 4))
|
|
self.assertEqual(self.set, set((2, 4)))
|
|
|
|
def test_intersection_superset(self):
|
|
self.set &= set([2, 4, 6, 8])
|
|
self.assertEqual(self.set, set([2, 4, 6]))
|
|
|
|
def test_intersection_overlap(self):
|
|
self.set &= set([3, 4, 5])
|
|
self.assertEqual(self.set, set([4]))
|
|
|
|
def test_intersection_non_overlap(self):
|
|
self.set &= set([8])
|
|
self.assertEqual(self.set, empty_set)
|
|
|
|
def test_intersection_method_call(self):
|
|
self.set.intersection_update(set([3, 4, 5]))
|
|
self.assertEqual(self.set, set([4]))
|
|
|
|
def test_sym_difference_subset(self):
|
|
self.set ^= set((2, 4))
|
|
self.assertEqual(self.set, set([6]))
|
|
|
|
def test_sym_difference_superset(self):
|
|
self.set ^= set((2, 4, 6, 8))
|
|
self.assertEqual(self.set, set([8]))
|
|
|
|
def test_sym_difference_overlap(self):
|
|
self.set ^= set((3, 4, 5))
|
|
self.assertEqual(self.set, set([2, 3, 5, 6]))
|
|
|
|
def test_sym_difference_non_overlap(self):
|
|
self.set ^= set([8])
|
|
self.assertEqual(self.set, set([2, 4, 6, 8]))
|
|
|
|
def test_sym_difference_method_call(self):
|
|
self.set.symmetric_difference_update(set([3, 4, 5]))
|
|
self.assertEqual(self.set, set([2, 3, 5, 6]))
|
|
|
|
def test_difference_subset(self):
|
|
self.set -= set((2, 4))
|
|
self.assertEqual(self.set, set([6]))
|
|
|
|
def test_difference_superset(self):
|
|
self.set -= set((2, 4, 6, 8))
|
|
self.assertEqual(self.set, set([]))
|
|
|
|
def test_difference_overlap(self):
|
|
self.set -= set((3, 4, 5))
|
|
self.assertEqual(self.set, set([2, 6]))
|
|
|
|
def test_difference_non_overlap(self):
|
|
self.set -= set([8])
|
|
self.assertEqual(self.set, set([2, 4, 6]))
|
|
|
|
def test_difference_method_call(self):
|
|
self.set.difference_update(set([3, 4, 5]))
|
|
self.assertEqual(self.set, set([2, 6]))
|
|
|
|
#==============================================================================
|
|
|
|
class TestMutate(unittest.TestCase):
|
|
def setUp(self):
|
|
self.values = ["a", "b", "c"]
|
|
self.set = set(self.values)
|
|
|
|
def test_add_present(self):
|
|
self.set.add("c")
|
|
self.assertEqual(self.set, set("abc"))
|
|
|
|
def test_add_absent(self):
|
|
self.set.add("d")
|
|
self.assertEqual(self.set, set("abcd"))
|
|
|
|
def test_add_until_full(self):
|
|
tmp = set()
|
|
expected_len = 0
|
|
for v in self.values:
|
|
tmp.add(v)
|
|
expected_len += 1
|
|
self.assertEqual(len(tmp), expected_len)
|
|
self.assertEqual(tmp, self.set)
|
|
|
|
def test_remove_present(self):
|
|
self.set.remove("b")
|
|
self.assertEqual(self.set, set("ac"))
|
|
|
|
def test_remove_absent(self):
|
|
try:
|
|
self.set.remove("d")
|
|
self.fail("Removing missing element should have raised LookupError")
|
|
except LookupError:
|
|
pass
|
|
|
|
def test_remove_until_empty(self):
|
|
expected_len = len(self.set)
|
|
for v in self.values:
|
|
self.set.remove(v)
|
|
expected_len -= 1
|
|
self.assertEqual(len(self.set), expected_len)
|
|
|
|
def test_discard_present(self):
|
|
self.set.discard("c")
|
|
self.assertEqual(self.set, set("ab"))
|
|
|
|
def test_discard_absent(self):
|
|
self.set.discard("d")
|
|
self.assertEqual(self.set, set("abc"))
|
|
|
|
def test_clear(self):
|
|
self.set.clear()
|
|
self.assertEqual(len(self.set), 0)
|
|
|
|
def test_pop(self):
|
|
popped = {}
|
|
while self.set:
|
|
popped[self.set.pop()] = None
|
|
self.assertEqual(len(popped), len(self.values))
|
|
for v in self.values:
|
|
self.assertIn(v, popped)
|
|
|
|
def test_update_empty_tuple(self):
|
|
self.set.update(())
|
|
self.assertEqual(self.set, set(self.values))
|
|
|
|
def test_update_unit_tuple_overlap(self):
|
|
self.set.update(("a",))
|
|
self.assertEqual(self.set, set(self.values))
|
|
|
|
def test_update_unit_tuple_non_overlap(self):
|
|
self.set.update(("a", "z"))
|
|
self.assertEqual(self.set, set(self.values + ["z"]))
|
|
|
|
#==============================================================================
|
|
|
|
class TestSubsets:
|
|
|
|
case2method = {"<=": "issubset",
|
|
">=": "issuperset",
|
|
}
|
|
|
|
reverse = {"==": "==",
|
|
"!=": "!=",
|
|
"<": ">",
|
|
">": "<",
|
|
"<=": ">=",
|
|
">=": "<=",
|
|
}
|
|
|
|
def test_issubset(self):
|
|
x = self.left
|
|
y = self.right
|
|
for case in "!=", "==", "<", "<=", ">", ">=":
|
|
expected = case in self.cases
|
|
# Test the binary infix spelling.
|
|
result = eval("x" + case + "y", locals())
|
|
self.assertEqual(result, expected)
|
|
# Test the "friendly" method-name spelling, if one exists.
|
|
if case in TestSubsets.case2method:
|
|
method = getattr(x, TestSubsets.case2method[case])
|
|
result = method(y)
|
|
self.assertEqual(result, expected)
|
|
|
|
# Now do the same for the operands reversed.
|
|
rcase = TestSubsets.reverse[case]
|
|
result = eval("y" + rcase + "x", locals())
|
|
self.assertEqual(result, expected)
|
|
if rcase in TestSubsets.case2method:
|
|
method = getattr(y, TestSubsets.case2method[rcase])
|
|
result = method(x)
|
|
self.assertEqual(result, expected)
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestSubsetEqualEmpty(TestSubsets, unittest.TestCase):
|
|
left = set()
|
|
right = set()
|
|
name = "both empty"
|
|
cases = "==", "<=", ">="
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestSubsetEqualNonEmpty(TestSubsets, unittest.TestCase):
|
|
left = set([1, 2])
|
|
right = set([1, 2])
|
|
name = "equal pair"
|
|
cases = "==", "<=", ">="
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestSubsetEmptyNonEmpty(TestSubsets, unittest.TestCase):
|
|
left = set()
|
|
right = set([1, 2])
|
|
name = "one empty, one non-empty"
|
|
cases = "!=", "<", "<="
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestSubsetPartial(TestSubsets, unittest.TestCase):
|
|
left = set([1])
|
|
right = set([1, 2])
|
|
name = "one a non-empty proper subset of other"
|
|
cases = "!=", "<", "<="
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestSubsetNonOverlap(TestSubsets, unittest.TestCase):
|
|
left = set([1])
|
|
right = set([2])
|
|
name = "neither empty, neither contains"
|
|
cases = "!="
|
|
|
|
#==============================================================================
|
|
|
|
class TestOnlySetsInBinaryOps:
|
|
|
|
def test_eq_ne(self):
|
|
# Unlike the others, this is testing that == and != *are* allowed.
|
|
self.assertEqual(self.other == self.set, False)
|
|
self.assertEqual(self.set == self.other, False)
|
|
self.assertEqual(self.other != self.set, True)
|
|
self.assertEqual(self.set != self.other, True)
|
|
|
|
def test_ge_gt_le_lt(self):
|
|
self.assertRaises(TypeError, lambda: self.set < self.other)
|
|
self.assertRaises(TypeError, lambda: self.set <= self.other)
|
|
self.assertRaises(TypeError, lambda: self.set > self.other)
|
|
self.assertRaises(TypeError, lambda: self.set >= self.other)
|
|
|
|
self.assertRaises(TypeError, lambda: self.other < self.set)
|
|
self.assertRaises(TypeError, lambda: self.other <= self.set)
|
|
self.assertRaises(TypeError, lambda: self.other > self.set)
|
|
self.assertRaises(TypeError, lambda: self.other >= self.set)
|
|
|
|
def test_update_operator(self):
|
|
try:
|
|
self.set |= self.other
|
|
except TypeError:
|
|
pass
|
|
else:
|
|
self.fail("expected TypeError")
|
|
|
|
def test_update(self):
|
|
if self.otherIsIterable:
|
|
self.set.update(self.other)
|
|
else:
|
|
self.assertRaises(TypeError, self.set.update, self.other)
|
|
|
|
def test_union(self):
|
|
self.assertRaises(TypeError, lambda: self.set | self.other)
|
|
self.assertRaises(TypeError, lambda: self.other | self.set)
|
|
if self.otherIsIterable:
|
|
self.set.union(self.other)
|
|
else:
|
|
self.assertRaises(TypeError, self.set.union, self.other)
|
|
|
|
def test_intersection_update_operator(self):
|
|
try:
|
|
self.set &= self.other
|
|
except TypeError:
|
|
pass
|
|
else:
|
|
self.fail("expected TypeError")
|
|
|
|
def test_intersection_update(self):
|
|
if self.otherIsIterable:
|
|
self.set.intersection_update(self.other)
|
|
else:
|
|
self.assertRaises(TypeError,
|
|
self.set.intersection_update,
|
|
self.other)
|
|
|
|
def test_intersection(self):
|
|
self.assertRaises(TypeError, lambda: self.set & self.other)
|
|
self.assertRaises(TypeError, lambda: self.other & self.set)
|
|
if self.otherIsIterable:
|
|
self.set.intersection(self.other)
|
|
else:
|
|
self.assertRaises(TypeError, self.set.intersection, self.other)
|
|
|
|
def test_sym_difference_update_operator(self):
|
|
try:
|
|
self.set ^= self.other
|
|
except TypeError:
|
|
pass
|
|
else:
|
|
self.fail("expected TypeError")
|
|
|
|
def test_sym_difference_update(self):
|
|
if self.otherIsIterable:
|
|
self.set.symmetric_difference_update(self.other)
|
|
else:
|
|
self.assertRaises(TypeError,
|
|
self.set.symmetric_difference_update,
|
|
self.other)
|
|
|
|
def test_sym_difference(self):
|
|
self.assertRaises(TypeError, lambda: self.set ^ self.other)
|
|
self.assertRaises(TypeError, lambda: self.other ^ self.set)
|
|
if self.otherIsIterable:
|
|
self.set.symmetric_difference(self.other)
|
|
else:
|
|
self.assertRaises(TypeError, self.set.symmetric_difference, self.other)
|
|
|
|
def test_difference_update_operator(self):
|
|
try:
|
|
self.set -= self.other
|
|
except TypeError:
|
|
pass
|
|
else:
|
|
self.fail("expected TypeError")
|
|
|
|
def test_difference_update(self):
|
|
if self.otherIsIterable:
|
|
self.set.difference_update(self.other)
|
|
else:
|
|
self.assertRaises(TypeError,
|
|
self.set.difference_update,
|
|
self.other)
|
|
|
|
def test_difference(self):
|
|
self.assertRaises(TypeError, lambda: self.set - self.other)
|
|
self.assertRaises(TypeError, lambda: self.other - self.set)
|
|
if self.otherIsIterable:
|
|
self.set.difference(self.other)
|
|
else:
|
|
self.assertRaises(TypeError, self.set.difference, self.other)
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestOnlySetsNumeric(TestOnlySetsInBinaryOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set((1, 2, 3))
|
|
self.other = 19
|
|
self.otherIsIterable = False
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestOnlySetsDict(TestOnlySetsInBinaryOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set((1, 2, 3))
|
|
self.other = {1:2, 3:4}
|
|
self.otherIsIterable = True
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestOnlySetsOperator(TestOnlySetsInBinaryOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set((1, 2, 3))
|
|
self.other = operator.add
|
|
self.otherIsIterable = False
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestOnlySetsTuple(TestOnlySetsInBinaryOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set((1, 2, 3))
|
|
self.other = (2, 4, 6)
|
|
self.otherIsIterable = True
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestOnlySetsString(TestOnlySetsInBinaryOps, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set((1, 2, 3))
|
|
self.other = 'abc'
|
|
self.otherIsIterable = True
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestOnlySetsGenerator(TestOnlySetsInBinaryOps, unittest.TestCase):
|
|
def setUp(self):
|
|
def gen():
|
|
for i in range(0, 10, 2):
|
|
yield i
|
|
self.set = set((1, 2, 3))
|
|
self.other = gen()
|
|
self.otherIsIterable = True
|
|
|
|
#==============================================================================
|
|
|
|
class TestCopying:
|
|
|
|
def test_copy(self):
|
|
dup = self.set.copy()
|
|
dup_list = sorted(dup, key=repr)
|
|
set_list = sorted(self.set, key=repr)
|
|
self.assertEqual(len(dup_list), len(set_list))
|
|
for i in range(len(dup_list)):
|
|
self.assertTrue(dup_list[i] is set_list[i])
|
|
|
|
def test_deep_copy(self):
|
|
dup = copy.deepcopy(self.set)
|
|
##print type(dup), repr(dup)
|
|
dup_list = sorted(dup, key=repr)
|
|
set_list = sorted(self.set, key=repr)
|
|
self.assertEqual(len(dup_list), len(set_list))
|
|
for i in range(len(dup_list)):
|
|
self.assertEqual(dup_list[i], set_list[i])
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestCopyingEmpty(TestCopying, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set()
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestCopyingSingleton(TestCopying, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set(["hello"])
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestCopyingTriple(TestCopying, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set(["zero", 0, None])
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestCopyingTuple(TestCopying, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set([(1, 2)])
|
|
|
|
#------------------------------------------------------------------------------
|
|
|
|
class TestCopyingNested(TestCopying, unittest.TestCase):
|
|
def setUp(self):
|
|
self.set = set([((1, 2), (3, 4))])
|
|
|
|
#==============================================================================
|
|
|
|
class TestIdentities(unittest.TestCase):
|
|
def setUp(self):
|
|
self.a = set('abracadabra')
|
|
self.b = set('alacazam')
|
|
|
|
def test_binopsVsSubsets(self):
|
|
a, b = self.a, self.b
|
|
self.assertTrue(a - b < a)
|
|
self.assertTrue(b - a < b)
|
|
self.assertTrue(a & b < a)
|
|
self.assertTrue(a & b < b)
|
|
self.assertTrue(a | b > a)
|
|
self.assertTrue(a | b > b)
|
|
self.assertTrue(a ^ b < a | b)
|
|
|
|
def test_commutativity(self):
|
|
a, b = self.a, self.b
|
|
self.assertEqual(a&b, b&a)
|
|
self.assertEqual(a|b, b|a)
|
|
self.assertEqual(a^b, b^a)
|
|
if a != b:
|
|
self.assertNotEqual(a-b, b-a)
|
|
|
|
def test_summations(self):
|
|
# check that sums of parts equal the whole
|
|
a, b = self.a, self.b
|
|
self.assertEqual((a-b)|(a&b)|(b-a), a|b)
|
|
self.assertEqual((a&b)|(a^b), a|b)
|
|
self.assertEqual(a|(b-a), a|b)
|
|
self.assertEqual((a-b)|b, a|b)
|
|
self.assertEqual((a-b)|(a&b), a)
|
|
self.assertEqual((b-a)|(a&b), b)
|
|
self.assertEqual((a-b)|(b-a), a^b)
|
|
|
|
def test_exclusion(self):
|
|
# check that inverse operations show non-overlap
|
|
a, b, zero = self.a, self.b, set()
|
|
self.assertEqual((a-b)&b, zero)
|
|
self.assertEqual((b-a)&a, zero)
|
|
self.assertEqual((a&b)&(a^b), zero)
|
|
|
|
# Tests derived from test_itertools.py =======================================
|
|
|
|
def R(seqn):
|
|
'Regular generator'
|
|
for i in seqn:
|
|
yield i
|
|
|
|
class G:
|
|
'Sequence using __getitem__'
|
|
def __init__(self, seqn):
|
|
self.seqn = seqn
|
|
def __getitem__(self, i):
|
|
return self.seqn[i]
|
|
|
|
class I:
|
|
'Sequence using iterator protocol'
|
|
def __init__(self, seqn):
|
|
self.seqn = seqn
|
|
self.i = 0
|
|
def __iter__(self):
|
|
return self
|
|
def __next__(self):
|
|
if self.i >= len(self.seqn): raise StopIteration
|
|
v = self.seqn[self.i]
|
|
self.i += 1
|
|
return v
|
|
|
|
class Ig:
|
|
'Sequence using iterator protocol defined with a generator'
|
|
def __init__(self, seqn):
|
|
self.seqn = seqn
|
|
self.i = 0
|
|
def __iter__(self):
|
|
for val in self.seqn:
|
|
yield val
|
|
|
|
class X:
|
|
'Missing __getitem__ and __iter__'
|
|
def __init__(self, seqn):
|
|
self.seqn = seqn
|
|
self.i = 0
|
|
def __next__(self):
|
|
if self.i >= len(self.seqn): raise StopIteration
|
|
v = self.seqn[self.i]
|
|
self.i += 1
|
|
return v
|
|
|
|
class N:
|
|
'Iterator missing __next__()'
|
|
def __init__(self, seqn):
|
|
self.seqn = seqn
|
|
self.i = 0
|
|
def __iter__(self):
|
|
return self
|
|
|
|
class E:
|
|
'Test propagation of exceptions'
|
|
def __init__(self, seqn):
|
|
self.seqn = seqn
|
|
self.i = 0
|
|
def __iter__(self):
|
|
return self
|
|
def __next__(self):
|
|
3 // 0
|
|
|
|
class S:
|
|
'Test immediate stop'
|
|
def __init__(self, seqn):
|
|
pass
|
|
def __iter__(self):
|
|
return self
|
|
def __next__(self):
|
|
raise StopIteration
|
|
|
|
from itertools import chain
|
|
def L(seqn):
|
|
'Test multiple tiers of iterators'
|
|
return chain(map(lambda x:x, R(Ig(G(seqn)))))
|
|
|
|
class TestVariousIteratorArgs(unittest.TestCase):
|
|
|
|
def test_constructor(self):
|
|
for cons in (set, frozenset):
|
|
for s in ("123", "", range(1000), ('do', 1.2), range(2000,2200,5)):
|
|
for g in (G, I, Ig, S, L, R):
|
|
self.assertEqual(sorted(cons(g(s)), key=repr), sorted(g(s), key=repr))
|
|
self.assertRaises(TypeError, cons , X(s))
|
|
self.assertRaises(TypeError, cons , N(s))
|
|
self.assertRaises(ZeroDivisionError, cons , E(s))
|
|
|
|
def test_inline_methods(self):
|
|
s = set('november')
|
|
for data in ("123", "", range(1000), ('do', 1.2), range(2000,2200,5), 'december'):
|
|
for meth in (s.union, s.intersection, s.difference, s.symmetric_difference, s.isdisjoint):
|
|
for g in (G, I, Ig, L, R):
|
|
expected = meth(data)
|
|
actual = meth(g(data))
|
|
if isinstance(expected, bool):
|
|
self.assertEqual(actual, expected)
|
|
else:
|
|
self.assertEqual(sorted(actual, key=repr), sorted(expected, key=repr))
|
|
self.assertRaises(TypeError, meth, X(s))
|
|
self.assertRaises(TypeError, meth, N(s))
|
|
self.assertRaises(ZeroDivisionError, meth, E(s))
|
|
|
|
def test_inplace_methods(self):
|
|
for data in ("123", "", range(1000), ('do', 1.2), range(2000,2200,5), 'december'):
|
|
for methname in ('update', 'intersection_update',
|
|
'difference_update', 'symmetric_difference_update'):
|
|
for g in (G, I, Ig, S, L, R):
|
|
s = set('january')
|
|
t = s.copy()
|
|
getattr(s, methname)(list(g(data)))
|
|
getattr(t, methname)(g(data))
|
|
self.assertEqual(sorted(s, key=repr), sorted(t, key=repr))
|
|
|
|
self.assertRaises(TypeError, getattr(set('january'), methname), X(data))
|
|
self.assertRaises(TypeError, getattr(set('january'), methname), N(data))
|
|
self.assertRaises(ZeroDivisionError, getattr(set('january'), methname), E(data))
|
|
|
|
class bad_eq:
|
|
def __eq__(self, other):
|
|
if be_bad:
|
|
set2.clear()
|
|
raise ZeroDivisionError
|
|
return self is other
|
|
def __hash__(self):
|
|
return 0
|
|
|
|
class bad_dict_clear:
|
|
def __eq__(self, other):
|
|
if be_bad:
|
|
dict2.clear()
|
|
return self is other
|
|
def __hash__(self):
|
|
return 0
|
|
|
|
class TestWeirdBugs(unittest.TestCase):
|
|
def test_8420_set_merge(self):
|
|
# This used to segfault
|
|
global be_bad, set2, dict2
|
|
be_bad = False
|
|
set1 = {bad_eq()}
|
|
set2 = {bad_eq() for i in range(75)}
|
|
be_bad = True
|
|
self.assertRaises(ZeroDivisionError, set1.update, set2)
|
|
|
|
be_bad = False
|
|
set1 = {bad_dict_clear()}
|
|
dict2 = {bad_dict_clear(): None}
|
|
be_bad = True
|
|
set1.symmetric_difference_update(dict2)
|
|
|
|
def test_iter_and_mutate(self):
|
|
# Issue #24581
|
|
s = set(range(100))
|
|
s.clear()
|
|
s.update(range(100))
|
|
si = iter(s)
|
|
s.clear()
|
|
a = list(range(100))
|
|
s.update(range(100))
|
|
list(si)
|
|
|
|
def test_merge_and_mutate(self):
|
|
# gh-141805
|
|
class X:
|
|
def __hash__(self):
|
|
return hash(0)
|
|
def __eq__(self, o):
|
|
other.clear()
|
|
return False
|
|
|
|
other = set()
|
|
other = {X() for i in range(10)}
|
|
s = {0}
|
|
s.update(other)
|
|
|
|
def test_hash_collision_concurrent_add(self):
|
|
class X:
|
|
def __hash__(self):
|
|
return 0
|
|
class Y:
|
|
flag = False
|
|
def __hash__(self):
|
|
return 0
|
|
def __eq__(self, other):
|
|
if not self.flag:
|
|
self.flag = True
|
|
s.add(X())
|
|
return self is other
|
|
|
|
a = X()
|
|
s = set()
|
|
s.add(a)
|
|
s.add(X())
|
|
s.remove(a)
|
|
# Now the set contains a dummy entry followed by an entry
|
|
# for an object with hash 0.
|
|
s.add(Y())
|
|
# The following operations should not crash.
|
|
repr(s)
|
|
list(s)
|
|
set() | s
|
|
|
|
|
|
class TestOperationsMutating:
|
|
"""Regression test for bpo-46615"""
|
|
|
|
constructor1 = None
|
|
constructor2 = None
|
|
|
|
def make_sets_of_bad_objects(self):
|
|
class Bad:
|
|
def __eq__(self, other):
|
|
if not enabled:
|
|
return False
|
|
if randrange(20) == 0:
|
|
set1.clear()
|
|
if randrange(20) == 0:
|
|
set2.clear()
|
|
return bool(randrange(2))
|
|
def __hash__(self):
|
|
return randrange(2)
|
|
# Don't behave poorly during construction.
|
|
enabled = False
|
|
set1 = self.constructor1(Bad() for _ in range(randrange(50)))
|
|
set2 = self.constructor2(Bad() for _ in range(randrange(50)))
|
|
# Now start behaving poorly
|
|
enabled = True
|
|
return set1, set2
|
|
|
|
def check_set_op_does_not_crash(self, function):
|
|
for _ in range(100):
|
|
set1, set2 = self.make_sets_of_bad_objects()
|
|
try:
|
|
function(set1, set2)
|
|
except RuntimeError as e:
|
|
# Just make sure we don't crash here.
|
|
self.assertIn("changed size during iteration", str(e))
|
|
|
|
|
|
class TestBinaryOpsMutating(TestOperationsMutating):
|
|
|
|
def test_eq_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a == b)
|
|
|
|
def test_ne_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a != b)
|
|
|
|
def test_lt_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a < b)
|
|
|
|
def test_le_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a <= b)
|
|
|
|
def test_gt_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a > b)
|
|
|
|
def test_ge_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a >= b)
|
|
|
|
def test_and_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a & b)
|
|
|
|
def test_or_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a | b)
|
|
|
|
def test_sub_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a - b)
|
|
|
|
def test_xor_with_mutation(self):
|
|
self.check_set_op_does_not_crash(lambda a, b: a ^ b)
|
|
|
|
def test_iadd_with_mutation(self):
|
|
def f(a, b):
|
|
a &= b
|
|
self.check_set_op_does_not_crash(f)
|
|
|
|
def test_ior_with_mutation(self):
|
|
def f(a, b):
|
|
a |= b
|
|
self.check_set_op_does_not_crash(f)
|
|
|
|
def test_isub_with_mutation(self):
|
|
def f(a, b):
|
|
a -= b
|
|
self.check_set_op_does_not_crash(f)
|
|
|
|
def test_ixor_with_mutation(self):
|
|
def f(a, b):
|
|
a ^= b
|
|
self.check_set_op_does_not_crash(f)
|
|
|
|
def test_iteration_with_mutation(self):
|
|
def f1(a, b):
|
|
for x in a:
|
|
pass
|
|
for y in b:
|
|
pass
|
|
def f2(a, b):
|
|
for y in b:
|
|
pass
|
|
for x in a:
|
|
pass
|
|
def f3(a, b):
|
|
for x, y in zip(a, b):
|
|
pass
|
|
self.check_set_op_does_not_crash(f1)
|
|
self.check_set_op_does_not_crash(f2)
|
|
self.check_set_op_does_not_crash(f3)
|
|
|
|
|
|
class TestBinaryOpsMutating_Set_Set(TestBinaryOpsMutating, unittest.TestCase):
|
|
constructor1 = set
|
|
constructor2 = set
|
|
|
|
class TestBinaryOpsMutating_Subclass_Subclass(TestBinaryOpsMutating, unittest.TestCase):
|
|
constructor1 = SetSubclass
|
|
constructor2 = SetSubclass
|
|
|
|
class TestBinaryOpsMutating_Set_Subclass(TestBinaryOpsMutating, unittest.TestCase):
|
|
constructor1 = set
|
|
constructor2 = SetSubclass
|
|
|
|
class TestBinaryOpsMutating_Subclass_Set(TestBinaryOpsMutating, unittest.TestCase):
|
|
constructor1 = SetSubclass
|
|
constructor2 = set
|
|
|
|
|
|
class TestMethodsMutating(TestOperationsMutating):
|
|
|
|
def test_issubset_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.issubset)
|
|
|
|
def test_issuperset_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.issuperset)
|
|
|
|
def test_intersection_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.intersection)
|
|
|
|
def test_union_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.union)
|
|
|
|
def test_difference_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.difference)
|
|
|
|
def test_symmetric_difference_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.symmetric_difference)
|
|
|
|
def test_isdisjoint_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.isdisjoint)
|
|
|
|
def test_difference_update_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.difference_update)
|
|
|
|
def test_intersection_update_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.intersection_update)
|
|
|
|
def test_symmetric_difference_update_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.symmetric_difference_update)
|
|
|
|
def test_update_with_mutation(self):
|
|
self.check_set_op_does_not_crash(set.update)
|
|
|
|
|
|
class TestMethodsMutating_Set_Set(TestMethodsMutating, unittest.TestCase):
|
|
constructor1 = set
|
|
constructor2 = set
|
|
|
|
class TestMethodsMutating_Subclass_Subclass(TestMethodsMutating, unittest.TestCase):
|
|
constructor1 = SetSubclass
|
|
constructor2 = SetSubclass
|
|
|
|
class TestMethodsMutating_Set_Subclass(TestMethodsMutating, unittest.TestCase):
|
|
constructor1 = set
|
|
constructor2 = SetSubclass
|
|
|
|
class TestMethodsMutating_Subclass_Set(TestMethodsMutating, unittest.TestCase):
|
|
constructor1 = SetSubclass
|
|
constructor2 = set
|
|
|
|
class TestMethodsMutating_Set_Dict(TestMethodsMutating, unittest.TestCase):
|
|
constructor1 = set
|
|
constructor2 = dict.fromkeys
|
|
|
|
class TestMethodsMutating_Set_List(TestMethodsMutating, unittest.TestCase):
|
|
constructor1 = set
|
|
constructor2 = list
|
|
|
|
|
|
# Application tests (based on David Eppstein's graph recipes ====================================
|
|
|
|
def powerset(U):
|
|
"""Generates all subsets of a set or sequence U."""
|
|
U = iter(U)
|
|
try:
|
|
x = frozenset([next(U)])
|
|
for S in powerset(U):
|
|
yield S
|
|
yield S | x
|
|
except StopIteration:
|
|
yield frozenset()
|
|
|
|
def cube(n):
|
|
"""Graph of n-dimensional hypercube."""
|
|
singletons = [frozenset([x]) for x in range(n)]
|
|
return dict([(x, frozenset([x^s for s in singletons]))
|
|
for x in powerset(range(n))])
|
|
|
|
def linegraph(G):
|
|
"""Graph, the vertices of which are edges of G,
|
|
with two vertices being adjacent iff the corresponding
|
|
edges share a vertex."""
|
|
L = {}
|
|
for x in G:
|
|
for y in G[x]:
|
|
nx = [frozenset([x,z]) for z in G[x] if z != y]
|
|
ny = [frozenset([y,z]) for z in G[y] if z != x]
|
|
L[frozenset([x,y])] = frozenset(nx+ny)
|
|
return L
|
|
|
|
def faces(G):
|
|
'Return a set of faces in G. Where a face is a set of vertices on that face'
|
|
# currently limited to triangles,squares, and pentagons
|
|
f = set()
|
|
for v1, edges in G.items():
|
|
for v2 in edges:
|
|
for v3 in G[v2]:
|
|
if v1 == v3:
|
|
continue
|
|
if v1 in G[v3]:
|
|
f.add(frozenset([v1, v2, v3]))
|
|
else:
|
|
for v4 in G[v3]:
|
|
if v4 == v2:
|
|
continue
|
|
if v1 in G[v4]:
|
|
f.add(frozenset([v1, v2, v3, v4]))
|
|
else:
|
|
for v5 in G[v4]:
|
|
if v5 == v3 or v5 == v2:
|
|
continue
|
|
if v1 in G[v5]:
|
|
f.add(frozenset([v1, v2, v3, v4, v5]))
|
|
return f
|
|
|
|
|
|
class TestGraphs(unittest.TestCase):
|
|
|
|
def test_cube(self):
|
|
|
|
g = cube(3) # vert --> {v1, v2, v3}
|
|
vertices1 = set(g)
|
|
self.assertEqual(len(vertices1), 8) # eight vertices
|
|
for edge in g.values():
|
|
self.assertEqual(len(edge), 3) # each vertex connects to three edges
|
|
vertices2 = set(v for edges in g.values() for v in edges)
|
|
self.assertEqual(vertices1, vertices2) # edge vertices in original set
|
|
|
|
cubefaces = faces(g)
|
|
self.assertEqual(len(cubefaces), 6) # six faces
|
|
for face in cubefaces:
|
|
self.assertEqual(len(face), 4) # each face is a square
|
|
|
|
def test_cuboctahedron(self):
|
|
|
|
# http://en.wikipedia.org/wiki/Cuboctahedron
|
|
# 8 triangular faces and 6 square faces
|
|
# 12 identical vertices each connecting a triangle and square
|
|
|
|
g = cube(3)
|
|
cuboctahedron = linegraph(g) # V( --> {V1, V2, V3, V4}
|
|
self.assertEqual(len(cuboctahedron), 12)# twelve vertices
|
|
|
|
vertices = set(cuboctahedron)
|
|
for edges in cuboctahedron.values():
|
|
self.assertEqual(len(edges), 4) # each vertex connects to four other vertices
|
|
othervertices = set(edge for edges in cuboctahedron.values() for edge in edges)
|
|
self.assertEqual(vertices, othervertices) # edge vertices in original set
|
|
|
|
cubofaces = faces(cuboctahedron)
|
|
facesizes = collections.defaultdict(int)
|
|
for face in cubofaces:
|
|
facesizes[len(face)] += 1
|
|
self.assertEqual(facesizes[3], 8) # eight triangular faces
|
|
self.assertEqual(facesizes[4], 6) # six square faces
|
|
|
|
for vertex in cuboctahedron:
|
|
edge = vertex # Cuboctahedron vertices are edges in Cube
|
|
self.assertEqual(len(edge), 2) # Two cube vertices define an edge
|
|
for cubevert in edge:
|
|
self.assertIn(cubevert, g)
|
|
|
|
|
|
#==============================================================================
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|