def test_keyset_multiplier(self):
params = Params()
params.keysets = {
'last_name_keyset':
dict(
key_parts=['_keyset', 'last_name'],
value_parts=['count'],
),
'first_name_keyset':
dict(
key_parts=['_keyset', 'first_name'],
value_parts=['count'],
)
}
data = [{
'first_name': 'Willow',
'last_name': 'Harvey'
}, {
'first_name': 'Noam',
'last_name': 'Clarke'
}]
expected = [{
'first_name': 'Willow',
'last_name': 'Harvey',
'_keyset': 'first_name_keyset'
}, {
'first_name': 'Willow',
'last_name': 'Harvey',
'_keyset': 'last_name_keyset'
}, {
'first_name': 'Noam',
'last_name': 'Clarke',
'_keyset': 'first_name_keyset'
}, {
'first_name': 'Noam',
'last_name': 'Clarke',
'_keyset': 'last_name_keyset'
}]
actual = keyset_multiplier(data, None, None, params)
ok_(isinstance(actual, types.GeneratorType))
eq_(list(actual), expected)
def test_keyset_multiplier(self):
params = Params()
params.keysets = {
'last_name_keyset': dict(
key_parts=['_keyset', 'last_name'],
value_parts=['count'],
),
'first_name_keyset': dict(
key_parts=['_keyset', 'first_name'],
value_parts=['count'],
)}
data = [
{'first_name': 'Willow', 'last_name': 'Harvey'},
{'first_name': 'Noam', 'last_name': 'Clarke'}]
expected = [
{
'first_name': 'Willow',
'last_name': 'Harvey',
'_keyset': 'first_name_keyset'
},
{
'first_name': 'Willow',
'last_name': 'Harvey',
'_keyset': 'last_name_keyset'
},
{
'first_name': 'Noam',
'last_name': 'Clarke',
'_keyset': 'first_name_keyset'
},
{
'first_name': 'Noam',
'last_name': 'Clarke',
'_keyset': 'last_name_keyset'
}]
actual = keyset_multiplier(data, None, None, params)
ok_(isinstance(actual, types.GeneratorType))
eq_(list(actual), expected)
