33
31
def load_tests(standard_tests, module, loader):
34
32
"""Parameterize tests for all versions of groupcompress."""
35
global compiled_static_tuple_feature
36
suite, compiled_static_tuple_feature = tests.permute_tests_for_extension(
37
standard_tests, loader, 'bzrlib._static_tuple_py',
38
'bzrlib._static_tuple_c')
34
('python', {'module': _static_tuple_py}),
36
suite = loader.suiteClass()
37
if CompiledStaticTuple.available():
38
from bzrlib import _static_tuple_c
39
scenarios.append(('C', {'module': _static_tuple_c}))
41
# the compiled module isn't available, so we add a failing test
42
class FailWithoutFeature(tests.TestCase):
44
self.requireFeature(CompiledStaticTuple)
45
suite.addTest(loader.loadTestsFromTestCase(FailWithoutFeature))
46
result = tests.multiply_tests(standard_tests, scenarios, suite)
50
class _CompiledStaticTuple(tests.Feature):
54
import bzrlib._static_tuple_c
59
def feature_name(self):
60
return 'bzrlib._static_tuple_c'
62
CompiledStaticTuple = _CompiledStaticTuple()
42
65
class _Meliae(tests.Feature):
77
100
def test_create_bad_args(self):
78
101
args_256 = ['a']*256
80
self.assertRaises(TypeError, self.module.StaticTuple, *args_256)
103
self.assertRaises(ValueError, self.module.StaticTuple, *args_256)
81
104
args_300 = ['a']*300
82
self.assertRaises(TypeError, self.module.StaticTuple, *args_300)
105
self.assertRaises(ValueError, self.module.StaticTuple, *args_300)
84
107
self.assertRaises(TypeError, self.module.StaticTuple, object())
123
146
k = self.module.StaticTuple('foo')
125
148
self.assertEqual(('foo',), t)
126
self.assertIsInstance(t, tuple)
127
self.assertFalse(isinstance(t, self.module.StaticTuple))
128
149
k = self.module.StaticTuple('foo', 'bar')
130
151
self.assertEqual(('foo', 'bar'), t)
131
k2 = self.module.StaticTuple(1, k)
133
self.assertIsInstance(t, tuple)
134
# For pickling to work, we need to keep the sub-items as StaticTuple so
135
# that it knows that they also need to be converted.
136
self.assertIsInstance(t[1], self.module.StaticTuple)
137
self.assertEqual((1, ('foo', 'bar')), t)
139
def test_as_tuples(self):
140
k1 = self.module.StaticTuple('foo', 'bar')
141
t = static_tuple.as_tuples(k1)
142
self.assertIsInstance(t, tuple)
143
self.assertEqual(('foo', 'bar'), t)
144
k2 = self.module.StaticTuple(1, k1)
145
t = static_tuple.as_tuples(k2)
146
self.assertIsInstance(t, tuple)
147
self.assertIsInstance(t[1], tuple)
148
self.assertEqual((1, ('foo', 'bar')), t)
150
t = static_tuple.as_tuples(mixed)
151
self.assertIsInstance(t, tuple)
152
self.assertIsInstance(t[1], tuple)
153
self.assertEqual((1, ('foo', 'bar')), t)
155
153
def test_len(self):
156
154
k = self.module.StaticTuple()
572
570
def test_from_sequence_not_sequence(self):
573
571
self.assertRaises(TypeError,
574
572
self.module.StaticTuple.from_sequence, object())
575
self.assertRaises(TypeError,
576
self.module.StaticTuple.from_sequence, 10)
578
574
def test_from_sequence_incorrect_args(self):
579
575
self.assertRaises(TypeError,
581
577
self.assertRaises(TypeError,
582
578
self.module.StaticTuple.from_sequence, foo='a')
584
def test_from_sequence_iterable(self):
585
st = self.module.StaticTuple.from_sequence(iter(['foo', 'bar']))
586
self.assertIsInstance(st, self.module.StaticTuple)
587
self.assertEqual(('foo', 'bar'), st)
589
def test_from_sequence_generator(self):
590
def generate_tuple():
593
st = self.module.StaticTuple.from_sequence(generate_tuple())
594
self.assertIsInstance(st, self.module.StaticTuple)
595
self.assertEqual(('foo', 'bar'), st)
597
def test_pickle(self):
598
st = self.module.StaticTuple('foo', 'bar')
599
pickled = cPickle.dumps(st)
600
unpickled = cPickle.loads(pickled)
601
self.assertEqual(unpickled, st)
603
def test_pickle_empty(self):
604
st = self.module.StaticTuple()
605
pickled = cPickle.dumps(st)
606
unpickled = cPickle.loads(pickled)
607
self.assertIs(st, unpickled)
609
def test_pickle_nested(self):
610
st = self.module.StaticTuple('foo', self.module.StaticTuple('bar'))
611
pickled = cPickle.dumps(st)
612
unpickled = cPickle.loads(pickled)
613
self.assertEqual(unpickled, st)
615
580
def test_static_tuple_thunk(self):
616
581
# Make sure the right implementation is available from
617
582
# bzrlib.static_tuple.StaticTuple.
618
583
if self.module is _static_tuple_py:
619
if compiled_static_tuple_feature.available():
584
if CompiledStaticTuple.available():
620
585
# We will be using the C version
622
587
self.assertIs(static_tuple.StaticTuple,
623
588
self.module.StaticTuple)
626
class TestEnsureStaticTuple(tests.TestCase):
628
def test_is_static_tuple(self):
629
st = static_tuple.StaticTuple('foo')
630
st2 = static_tuple.expect_static_tuple(st)
631
self.assertIs(st, st2)
633
def test_is_tuple(self):
635
st = static_tuple.expect_static_tuple(t)
636
self.assertIsInstance(st, static_tuple.StaticTuple)
637
self.assertEqual(t, st)
639
def test_flagged_is_static_tuple(self):
640
debug.debug_flags.add('static_tuple')
641
st = static_tuple.StaticTuple('foo')
642
st2 = static_tuple.expect_static_tuple(st)
643
self.assertIs(st, st2)
645
def test_flagged_is_tuple(self):
646
debug.debug_flags.add('static_tuple')
648
self.assertRaises(TypeError, static_tuple.expect_static_tuple, t)