summaryrefslogtreecommitdiff
path: root/tests/test_more.py
diff options
context:
space:
mode:
authorNathan Brooks <natebrooks004@gmail.com>2021-01-02 14:03:49 -0800
committerNathan Brooks <natebrooks004@gmail.com>2021-01-02 14:23:06 -0800
commit89d6a8e2e5b92ae0787c3254cec5f324da3362c8 (patch)
tree45b48f04f0bc425848aed46249b008f0b038b8aa /tests/test_more.py
parent64b99e9b183987260da17927dbf37bf719adf7ed (diff)
downloadmore-itertools-89d6a8e2e5b92ae0787c3254cec5f324da3362c8.tar.gz
add product_index, combination_index, and permutation_index functions
Diffstat (limited to 'tests/test_more.py')
-rw-r--r--tests/test_more.py30
1 files changed, 15 insertions, 15 deletions
diff --git a/tests/test_more.py b/tests/test_more.py
index 665ee07..7ac88e4 100644
--- a/tests/test_more.py
+++ b/tests/test_more.py
@@ -4274,7 +4274,7 @@ class CombinationIndexTests(TestCase):
r = 4
first_index = {}
for index, element in enumerate(combinations(iterable, r)):
- actual = mi.combination_index(iterable, element)
+ actual = mi.combination_index(element, iterable)
expected = first_index.setdefault(element, index)
self.assertEqual(actual, expected)
@@ -4283,7 +4283,7 @@ class CombinationIndexTests(TestCase):
r = len(iterable)
first_index = {}
for index, element in enumerate(combinations(iterable, r=r)):
- actual = mi.permutation_index(iterable, element)
+ actual = mi.combination_index(element, iterable)
expected = first_index.setdefault(element, index)
self.assertEqual(actual, expected)
@@ -4292,31 +4292,31 @@ class CombinationIndexTests(TestCase):
r = 3
first_index = {}
for index, element in enumerate(combinations(iterable, r)):
- actual = mi.combination_index(iterable, element)
+ actual = mi.combination_index(element, iterable)
expected = first_index.setdefault(element, index)
self.assertEqual(actual, expected)
def test_null(self):
- actual = mi.combination_index([], tuple())
+ actual = mi.combination_index(tuple(), [])
expected = 0
self.assertEqual(actual, expected)
def test_long(self):
- actual = mi.combination_index(range(180), (2, 12, 35, 126))
+ actual = mi.combination_index((2, 12, 35, 126), range(180))
expected = 2000000
self.assertEqual(actual, expected)
def test_invalid_order(self):
with self.assertRaises(ValueError):
- mi.combination_index('abcde', tuple('acb'))
+ mi.combination_index(tuple('acb'), 'abcde')
def test_invalid_large(self):
with self.assertRaises(ValueError):
- mi.combination_index('abcdef', tuple('abcdefg'))
+ mi.combination_index(tuple('abcdefg'), 'abcdef')
def test_invalid_match(self):
with self.assertRaises(ValueError):
- mi.product_index('abcde', tuple('axe'))
+ mi.product_index(tuple('axe'), 'abcde')
class PermutationIndexTests(TestCase):
@@ -4325,7 +4325,7 @@ class PermutationIndexTests(TestCase):
r = 4
first_index = {}
for index, element in enumerate(permutations(iterable, r)):
- actual = mi.permutation_index(iterable, element)
+ actual = mi.permutation_index(element, iterable)
expected = first_index.setdefault(element, index)
self.assertEqual(actual, expected)
@@ -4333,7 +4333,7 @@ class PermutationIndexTests(TestCase):
iterable = 'abcd'
first_index = {}
for index, element in enumerate(permutations(iterable)):
- actual = mi.permutation_index(iterable, element)
+ actual = mi.permutation_index(element, iterable)
expected = first_index.setdefault(element, index)
self.assertEqual(actual, expected)
@@ -4342,24 +4342,24 @@ class PermutationIndexTests(TestCase):
r = 3
first_index = {}
for index, element in enumerate(permutations(iterable, r)):
- actual = mi.permutation_index(iterable, element)
+ actual = mi.permutation_index(element, iterable)
expected = first_index.setdefault(element, index)
self.assertEqual(actual, expected)
def test_null(self):
- actual = mi.permutation_index([], tuple())
+ actual = mi.permutation_index(tuple(), [])
expected = 0
self.assertEqual(actual, expected)
def test_long(self):
- actual = mi.permutation_index(range(180), (2, 12, 35, 126))
+ actual = mi.permutation_index((2, 12, 35, 126), range(180))
expected = 11631678
self.assertEqual(actual, expected)
def test_invalid_large(self):
with self.assertRaises(ValueError):
- mi.permutation_index('abcdef', tuple('abcdefg'))
+ mi.permutation_index(tuple('abcdefg'), 'abcdef')
def test_invalid_match(self):
with self.assertRaises(ValueError):
- mi.product_index('abcde', tuple('axe'))
+ mi.product_index(tuple('axe'), 'abcde')