def setUpHatefulComment(self): """ Custom setup """ self.annotators = sorted(UserFactory.create_batch(5), key=lambda x: x.username) label_article(self.annotators[0], self.article, is_interesting=True, comments=[ ["HATE", "MUJER"], [], [], ]) label_article(self.annotators[1], self.article, is_interesting=True, comments=[ ["HATE", "MUJER", "ASPECTO"], [], [], ]) label_article(self.annotators[3], self.article, is_interesting=True, comments=[ [], [], ["HATE", "OTROS"], ])
def setUp(self): """ Setup """ self.article = ArticleFactory(create_comments__num_comments=3) self.comments = self.article.comment_set.all() self.annotators = sorted(UserFactory.create_batch(3), key=lambda x: x.username)
def test_non_hateful_comment(self): """ Get first assignment """ annotators = sorted(UserFactory.create_batch(5), key=lambda x: x.username) for annotator in annotators: label_article(annotator, self.article, is_interesting=True, comments=[ [], [], [], ]) serializer = CommentSerializer(raw=True) for comment in self.article.comment_set.all(): serialized_comment = serializer.serialize(comment) for key in field_mapping: self.assertCountEqual(serialized_comment[key], [])