def testGivenKeyFunctionAsPositionalAttributeGroupsByItsResult(self):
self.assertEqual(
{
2: [(0, 2), (1, 1), (1, 1)],
3: [(1, 2), (2, 1), (3, 0)]
},
groupBy([(1, 2), (2, 1), (0, 2), (3, 0), (1, 1), (1, 1)],
lambda x: x[0] + x[1]))
def testGivenKeyFunctionAsKeywordAttributeGroupsByItsResult(self):
self.assertEqual(
{
1: [(1, 1), (2, 1, 8)],
2: [(1, 2), (3, 2)],
4: [(3, 4)]
},
groupBy([(1, 2), (3, 4), (3, 2), (1, 1), (2, 1, 8)],
getKey=operator.itemgetter(1)))
def testKeyFunctionDefaultsToIdentity(self):
self.assertEqual({1: [1],
2: [2]},
groupBy([1, 2]))
def testGivenTupleOfNamesOfAttributesGroupsByTheirCombination(self):
self.assertEqual({(1, 1): [(1, 1)],
(1, 2): [(1, 2)],
(2, 1): [(2, 1)]},
groupBy([Pair(1, 2), Pair(1, 1), Pair(2, 1)], ('a', 'b')))
def testGivenNameOfAttributeGroupsByTheAttribute(self):
self.assertEqual({1: [(1, 2), (1, 1)],
2: [(2, 1)]},
groupBy([Pair(1, 2), Pair(1, 1), Pair(2, 1)], 'a'))
def testGivenNoObjectsReturnsEmptyDict(self):
self.assertEqual({},
groupBy([], getKey=lambda x: x[0] + x[1]))
def testGivenTypeAsKeyFunctionGroupsByItsResult(self):
self.assertEqual({'2': [2, 2],
'3': [3]},
groupBy([2, 3, 2],
getKey=str))
def testGivenKeyFunctionAsPositionalAttributeGroupsByItsResult(self):
self.assertEqual({2: [(0, 2), (1, 1), (1, 1)],
3: [(1, 2), (2, 1), (3, 0)]},
groupBy([(1, 2), (2, 1), (0, 2), (3, 0), (1, 1), (1, 1)],
lambda x: x[0] + x[1]))
def testRequiredKeysDoNotConflictWithKeysDefinedByInput(self):
self.assertEqual({1: [1]}, groupBy([1], requiredKeys=[1]))
def testOutputAlwaysContainsRequiredKeys(self):
self.assertEqual({1: [1], 'a': []}, groupBy([1], requiredKeys=['a']))
def testKeyFunctionDefaultsToIdentity(self):
self.assertEqual({1: [1], 2: [2]}, groupBy([1, 2]))
def testGivenTupleOfNamesOfAttributesGroupsByTheirCombination(self):
self.assertEqual({
(1, 1): [(1, 1)],
(1, 2): [(1, 2)],
(2, 1): [(2, 1)]
}, groupBy([Pair(1, 2), Pair(1, 1), Pair(2, 1)], ('a', 'b')))
def testGivenNameOfAttributeGroupsByTheAttribute(self):
self.assertEqual({
1: [(1, 2), (1, 1)],
2: [(2, 1)]
}, groupBy([Pair(1, 2), Pair(1, 1), Pair(2, 1)], 'a'))
def testGivenNoObjectsReturnsEmptyDict(self):
self.assertEqual({}, groupBy([], getKey=lambda x: x[0] + x[1]))
def testGivenTypeAsKeyFunctionGroupsByItsResult(self):
self.assertEqual({
'2': [2, 2],
'3': [3]
}, groupBy([2, 3, 2], getKey=str))
def testOutputAlwaysContainsRequiredKeys(self):
self.assertEqual({1: [1],
'a': []},
groupBy([1], requiredKeys=['a']))
def testRequiredKeysDoNotConflictWithKeysDefinedByInput(self):
self.assertEqual({1: [1]},
groupBy([1], requiredKeys=[1]))
def testGivenKeyFunctionAsKeywordAttributeGroupsByItsResult(self):
self.assertEqual({1: [(1, 1), (2, 1, 8)],
2: [(1, 2), (3, 2)],
4: [(3, 4)]},
groupBy([(1, 2), (3, 4), (3, 2), (1, 1), (2, 1, 8)],
getKey=operator.itemgetter(1)))