def test_hash_item_key(self):
cases = [
(lambda x: x, "64ad3fb166ddb41a2ca24f1803b8b722"),
(lambda x: x.upper(), "ee22e8597bff91742affe4befbf4649a"),
]
for key, expected in cases:
with self.subTest(key=key):
shuffler = Shuffler(seed=1234)
actual = shuffler._hash_item("abc", key)
self.assertEqual(actual, expected)
def test_shuffle_key(self):
cases = [
(lambda x: x, ["a", "d", "b", "c"]),
(lambda x: x.upper(), ["d", "c", "a", "b"]),
]
for num, (key, expected) in enumerate(cases, start=1):
with self.subTest(num=num):
shuffler = Shuffler(seed=1234)
actual = shuffler.shuffle(["a", "b", "c", "d"], key)
self.assertEqual(actual, expected)
def test_shuffle_key(self):
cases = [
(lambda x: x, ['a', 'd', 'b', 'c']),
(lambda x: x.upper(), ['d', 'c', 'a', 'b']),
]
for num, (key, expected) in enumerate(cases, start=1):
with self.subTest(num=num):
shuffler = Shuffler(seed=1234)
actual = shuffler.shuffle(['a', 'b', 'c', 'd'], key)
self.assertEqual(actual, expected)
def test_hash_item_seed(self):
cases = [
(1234, "64ad3fb166ddb41a2ca24f1803b8b722"),
# Passing a string gives the same value.
("1234", "64ad3fb166ddb41a2ca24f1803b8b722"),
(5678, "4dde450ad339b6ce45a0a2666e35b975"),
]
for seed, expected in cases:
with self.subTest(seed=seed):
shuffler = Shuffler(seed=seed)
actual = shuffler._hash_item("abc", lambda x: x)
self.assertEqual(actual, expected)
def test_shuffle_consistency(self):
seq = [str(n) for n in range(5)]
cases = [
(None, ["3", "0", "2", "4", "1"]),
(0, ["3", "2", "4", "1"]),
(1, ["3", "0", "2", "4"]),
(2, ["3", "0", "4", "1"]),
(3, ["0", "2", "4", "1"]),
(4, ["3", "0", "2", "1"]),
]
shuffler = Shuffler(seed=1234)
for index, expected in cases:
with self.subTest(index=index):
if index is None:
new_seq = seq
else:
new_seq = seq.copy()
del new_seq[index]
actual = shuffler.shuffle(new_seq, lambda x: x)
self.assertEqual(actual, expected)
def test_reorder_tests_random(self):
tests = self.make_tests()
# Choose a seed that shuffles both the classes and methods.
shuffler = Shuffler(seed=9)
reordered_tests = reorder_tests(tests, classes=[], shuffler=shuffler)
self.assertIsInstance(reordered_tests, collections.abc.Iterator)
self.assertTestNames(reordered_tests,
expected=[
'Tests2.test1',
'Tests2.test2',
'Tests1.test2',
'Tests1.test1',
])
def test_reorder_test_bin_random_and_reverse(self):
tests = self.make_tests()
# Choose a seed that shuffles both the classes and methods.
shuffler = Shuffler(seed=9)
reordered_tests = reorder_test_bin(tests, shuffler=shuffler, reverse=True)
self.assertIsInstance(reordered_tests, collections.abc.Iterator)
self.assertTestNames(
reordered_tests,
expected=[
"Tests1.test1",
"Tests1.test2",
"Tests2.test2",
"Tests2.test1",
],
)
def test_shuffle_tests(self):
tests = self.make_tests()
# Choose a seed that shuffles both the classes and methods.
shuffler = Shuffler(seed=9)
shuffled_tests = shuffle_tests(tests, shuffler)
self.assertIsInstance(shuffled_tests, collections.abc.Iterator)
self.assertTestNames(
shuffled_tests,
expected=[
"Tests2.test1",
"Tests2.test2",
"Tests1.test2",
"Tests1.test1",
],
)
def test_reorder_tests_random_mixed_classes(self):
tests = self.make_tests()
# Move the last item to the front.
tests.insert(0, tests.pop())
shuffler = Shuffler(seed=9)
self.assertTestNames(tests,
expected=[
'Tests2.test2',
'Tests1.test1',
'Tests1.test2',
'Tests2.test1',
])
reordered_tests = reorder_tests(tests, classes=[], shuffler=shuffler)
self.assertTestNames(reordered_tests,
expected=[
'Tests2.test1',
'Tests2.test2',
'Tests1.test2',
'Tests1.test1',
])
def test_shuffle_same_hash(self):
shuffler = Shuffler(seed=1234)
msg = "item 'A' has same hash 'a56ce89262959e151ee2266552f1819c' as item 'a'"
with self.assertRaisesMessage(RuntimeError, msg):
shuffler.shuffle(["a", "b", "A"], lambda x: x.upper())
def test_hash_text(self):
actual = Shuffler._hash_text("abcd")
self.assertEqual(actual, "e2fc714c4727ee9395f324cd2e7f331f")
def test_seed_display(self):
shuffler = Shuffler(100)
shuffler.seed_source = "test"
self.assertEqual(shuffler.seed_display, "100 (test)")
def test_init_no_seed_argument(self):
with mock.patch("random.randint", return_value=300):
shuffler = Shuffler()
self.assertEqual(shuffler.seed, 300)
self.assertEqual(shuffler.seed_source, "generated")
def test_init_none_seed(self):
with mock.patch("random.randint", return_value=200):
shuffler = Shuffler(None)
self.assertEqual(shuffler.seed, 200)
self.assertEqual(shuffler.seed_source, "generated")
def test_init(self):
shuffler = Shuffler(100)
self.assertEqual(shuffler.seed, 100)
self.assertEqual(shuffler.seed_source, "given")
def test_seed_display(self):
shuffler = Shuffler(100)
shuffler.seed_source = 'test'
self.assertEqual(shuffler.seed_display, '100 (test)')
