def test_strict_tag_application(self):
self.setUpTags()
# test that when there are two options I select the actor name
tags = Tag.objects.filter(name='Actor Name')
items = NameContent.objects.filter(tags__tag__name='English')
self.assertEqual(len(tags), 1)
self.assertEqual(len(items), 2)
selection = select_random_from(items, tags, choices=1)
self.assertEqual(len(selection), 1)
self.assertTrue(isinstance(selection[0], NameContent))
self.assertEqual(selection[0].name, 'William')
# test that when there is one option, but it's not an actor name, I don't select it
items = items.filter(name='The Jolly Fisherman')
self.assertEqual(len(items), 1)
selection = select_random_from(items, tags, choices=1)
self.assertEqual(len(selection), 0)
def test_select_random_from_empty_set(self):
self.setUpTags()
t = Tag(name='Scottish', type=TagType.Culture)
t.save()
tags = Tag.objects.filter(name=t.name)
selection = select_random_from(NameContent.objects.all(), tags, superset_only=True)
self.assertEqual(len(selection), 0)
def test_select_random_from_superset_only(self):
self.setUpTags()
# choose two tags that only one NameContent has
tags = Tag.objects.filter(name='English')
english_names = NameContent.objects.filter(tags__tag__name='English')
english_names = english_names.filter(tags__tag__name='Actor Name')
self.assertEqual(len(english_names), 1)
selection = select_random_from(NameContent.objects.all(), tags, superset_only=True)
self.assertEqual(len(selection), 1)
self.assertTrue(isinstance(selection[0], NameContent))
self.assertEqual(selection[0].tags.all()[0].tag.name, 'English')
def test_select_random_from_multiple(self):
self.setUpTags()
selection = select_random_from(NameContent.objects.all(), Tag.objects.all(), choices=2)
self.assertEqual(len(selection), 2)
self.assertTrue(isinstance(selection[0], NameContent))