def setUp(self):
self.translations = [
{
'pl': 'kobieta',
'br': 'a mulher',
'en': 'woman',
'group': 'basics',
},
{
'pl': 'dziewczyna',
'br': 'a menina',
'en': 'girl',
'group': 'basics',
},
{
'pl': 'mężczyzna',
'br': 'o homem',
'en': 'man',
'group': 'basics2',
},
{
'pl': 'chłopiec',
'br': 'o menino',
'en': 'boy',
'group': 'basics2',
},
]
self.single_group_filter = GroupFilter({u'basics'})
self.unknown_group_filter = GroupFilter({u'unknown'})
self.multi_group_filter = GroupFilter({u'basics', u'basics2'})
def setUp(self):
self.translations = [
{
'pl': 'kobieta',
'br': 'a mulher',
'en': 'woman',
'group': 'basics',
},
{
'pl': 'dziewczyna',
'br': 'a menina',
'en': 'girl',
'group': 'basics',
},
{
'pl': 'mężczyzna',
'br': 'o homem',
'en': 'man',
'group': 'basics2',
},
{
'pl': 'chłopiec',
'br': 'o menino',
'en': 'boy',
'group': 'basics2',
},
]
self.mixed_filter = GroupFilter({u'basics2'}).link(CountFilter(1))
class MixedFilterTests(unittest.TestCase):
def setUp(self):
self.translations = [
{
'pl': 'kobieta',
'br': 'a mulher',
'en': 'woman',
'group': 'basics',
},
{
'pl': 'dziewczyna',
'br': 'a menina',
'en': 'girl',
'group': 'basics',
},
{
'pl': 'mężczyzna',
'br': 'o homem',
'en': 'man',
'group': 'basics2',
},
{
'pl': 'chłopiec',
'br': 'o menino',
'en': 'boy',
'group': 'basics2',
},
]
self.mixed_filter = GroupFilter({u'basics2'}).link(CountFilter(1))
def test_mixed_filters(self):
filtered = self.mixed_filter.filter(self.translations)
self.assertEqual(len(filtered), 1)
self.assertEqual(filtered, self.translations[-1:])
class GroupFilterTests(unittest.TestCase):
def setUp(self):
self.translations = [
{
'pl': 'kobieta',
'br': 'a mulher',
'en': 'woman',
'group': 'basics',
},
{
'pl': 'dziewczyna',
'br': 'a menina',
'en': 'girl',
'group': 'basics',
},
{
'pl': 'mężczyzna',
'br': 'o homem',
'en': 'man',
'group': 'basics2',
},
{
'pl': 'chłopiec',
'br': 'o menino',
'en': 'boy',
'group': 'basics2',
},
]
self.single_group_filter = GroupFilter({u'basics'})
self.unknown_group_filter = GroupFilter({u'unknown'})
self.multi_group_filter = GroupFilter({u'basics', u'basics2'})
def test_single_group(self):
filtered = self.single_group_filter.filter(self.translations)
self.assertEqual(len(filtered), 2)
self.assertEqual(filtered, self.translations[:2])
def test_unknown_group(self):
filtered = self.unknown_group_filter.filter(self.translations)
self.assertEqual(len(filtered), 0)
def test_multi_groups(self):
filtered = self.multi_group_filter.filter(self.translations)
self.assertEqual(len(filtered), len(self.translations))
self.assertEqual(filtered, self.translations)
