def test_lookup_apply_result_selector_default(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup = Lookup(k_v)
result = lookup.apply_result_selector().to_list()
# Rely on the fact that the Lookup is ordered by first key insertion
self.assertEqual(result[0], lookup['a'])
self.assertEqual(result[1], lookup['b'])
self.assertEqual(result[2], lookup['c'])
def test_lookup_apply_result_selector(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup = Lookup(k_v)
result = lookup.apply_result_selector(lambda key, seq: key).to_list()
self.assertEqual(len(result), 3)
self.assertTrue('a' in result)
self.assertTrue('b' in result)
self.assertTrue('c' in result)
def test_lookup_inequality_negative(self):
kv1 = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
group1 = Lookup(kv1)
kv2 = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
group2 = Lookup(kv2)
self.assertFalse(group1 != group2)
def test_lookup_inequality_positive(self):
kv1 = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
group1 = Lookup(kv1)
kv2 = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('s', 'star fruit')]
group2 = Lookup(kv2)
self.assertTrue(group1 != group2)
def test_lookup_to_dictionary(self):
kv1 = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup1 = Lookup(kv1)
actual = lookup1.to_dictionary(
key_selector=lambda grouping: len(grouping),
value_selector=lambda grouping: grouping.element_at(0))
expected = {1: 'artichoke', 2: 'blackberry', 3: 'clementine'}
self.assertEqual(actual, expected)
def test_lookup_to_dictionary_default_selectors(self):
kv1 = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup1 = Lookup(kv1)
actual = lookup1.to_dictionary()
expected = {
'a': ['artichoke'],
'b': ['blackberry', 'blueberry'],
'c': ['clementine', 'cranberry', 'cantaloupe']
}
self.assertEqual(actual, expected)
def test_lookup_to_dictionary_default_selectors(self):
kv1 = [ ('a', 'artichoke'),
('b', 'blackberry'),
('c', 'clementine'),
('b', 'blueberry'),
('c', 'cranberry'),
('c', 'cantaloupe') ]
lookup1 = Lookup(kv1)
actual = lookup1.to_dictionary()
expected = {'a': ['artichoke'],
'b': ['blackberry', 'blueberry'],
'c': ['clementine', 'cranberry', 'cantaloupe']}
self.assertEqual(actual, expected)
def test_lookup_apply_result_selector(self):
k_v = [ ('a', 'artichoke'),
('b', 'blackberry'),
('c', 'clementine'),
('b', 'blueberry'),
('c', 'cranberry'),
('c', 'cantaloupe') ]
lookup = Lookup(k_v)
result = lookup.apply_result_selector(lambda key, seq: key).to_list()
self.assertEqual(len(result), 3)
self.assertTrue('a' in result)
self.assertTrue('b' in result)
self.assertTrue('c' in result)
def test_lookup_is_ordered(self):
k_v = [ ('a', 'artichoke'),
('b', 'blackberry'),
('c', 'clementine'),
('b', 'blueberry'),
('c', 'cranberry'),
('c', 'cantaloupe') ]
lookup = Lookup(k_v)
result = lookup.apply_result_selector().to_list()
# Rely on the fact that the Lookup is ordered by first key insertion
self.assertEqual(list(result[0]), ['artichoke'])
self.assertEqual(list(result[1]), ['blackberry', 'blueberry'])
self.assertEqual(list(result[2]), ['clementine', 'cranberry', 'cantaloupe'])
def test_lookup_is_ordered(self):
k_v = [
("a", "artichoke"),
("b", "blackberry"),
("c", "clementine"),
("b", "blueberry"),
("c", "cranberry"),
("c", "cantaloupe"),
]
lookup = Lookup(k_v)
result = lookup.apply_result_selector().to_list()
# Rely on the fact that the Lookup is ordered by first key insertion
self.assertEqual(list(result[0]), ["artichoke"])
self.assertEqual(list(result[1]), ["blackberry", "blueberry"])
self.assertEqual(list(result[2]), ["clementine", "cranberry", "cantaloupe"])
def test_lookup_apply_result_selector(self):
k_v = [
("a", "artichoke"),
("b", "blackberry"),
("c", "clementine"),
("b", "blueberry"),
("c", "cranberry"),
("c", "cantaloupe"),
]
lookup = Lookup(k_v)
result = lookup.apply_result_selector(lambda key, seq: key).to_list()
self.assertEqual(len(result), 3)
self.assertTrue("a" in result)
self.assertTrue("b" in result)
self.assertTrue("c" in result)
def test_lookup_as_queryable(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
a_group = Lookup(k_v).where(lambda g: g.key == 'a').to_list()
self.assertEqual(len(a_group), 1)
self.assertTrue(isinstance(a_group[0], Grouping))
self.assertEqual(a_group[0].key, 'a')
def test_lookup_to_dictionary(self):
kv1 = [ ('a', 'artichoke'),
('b', 'blackberry'),
('c', 'clementine'),
('b', 'blueberry'),
('c', 'cranberry'),
('c', 'cantaloupe') ]
lookup1 = Lookup(kv1)
actual = lookup1.to_dictionary(
key_selector=lambda grouping: len(grouping),
value_selector=lambda grouping: grouping.element_at(0)
)
expected = {1: 'artichoke',
2: 'blackberry',
3: 'clementine'}
self.assertEqual(actual, expected)
def test_lookup_repr(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup = Lookup(k_v)
self.assertEqual(
repr(lookup),
"Lookup([('a', 'artichoke'), ('b', 'blackberry'), ('b', 'blueberry'), ('c', 'clementine'), ('c', 'cranberry'), ('c', 'cantaloupe')])"
)
def test_lookup_iterable(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup = Lookup(k_v)
count = 0
for group in lookup:
count += 1
self.assertTrue(isinstance(group, Grouping))
self.assertEqual(count, 3)
# Check that we can iterate again
count = 0
for group in lookup:
count += 1
self.assertTrue(isinstance(group, Grouping))
self.assertEqual(count, 3)
def test_lookup_getitem_positive(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup = Lookup(k_v)
g_a = lookup['a']
self.assertEqual(g_a.key, 'a')
self.assertTrue('artichoke' in g_a)
g_b = lookup['b']
self.assertEqual(g_b.key, 'b')
self.assertTrue('blackberry' in g_b)
self.assertTrue('blueberry' in g_b)
g_c = lookup['c']
self.assertEqual(g_c.key, 'c')
self.assertTrue('clementine' in g_c)
self.assertTrue('cranberry' in g_c)
self.assertTrue('cantaloupe' in g_c)
def test_lookup_create(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup = Lookup(k_v)
def test_lookup_in_positive(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', 'clementine'),
('b', 'blueberry'), ('c', 'cranberry'), ('c', 'cantaloupe')]
lookup = Lookup(k_v)
self.assertFalse('z' in lookup)
def test_lookup_create_empty(self):
k_v = []
lookup = Lookup(k_v)
def test_lookup_create_invalid(self):
k_v = [('a', 'artichoke'), ('b', 'blackberry'), ('c', ),
('b', 'blueberry'), ('c', ), ('c', 'cantaloupe')]
self.assertRaises(ValueError, lambda: Lookup(k_v))
