From 43b5d614830d6cd3c73c3baa65900d01b0f9ed6a Mon Sep 17 00:00:00 2001 From: Jonas Winkler Date: Tue, 17 Nov 2020 17:13:08 +0100 Subject: [PATCH] codestyle changes --- src/documents/tests/test_classifier.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/documents/tests/test_classifier.py b/src/documents/tests/test_classifier.py index 1dc859fdb..9c85b786a 100644 --- a/src/documents/tests/test_classifier.py +++ b/src/documents/tests/test_classifier.py @@ -48,21 +48,19 @@ class TestClassifier(TestCase): self.assertIsNone(self.classifier.predict_document_type("")) self.assertIsNone(self.classifier.predict_correspondent("")) - def testTrain(self): self.generate_test_data() self.classifier.train() self.assertListEqual(list(self.classifier.correspondent_classifier.classes_), [-1, self.c1.pk]) self.assertListEqual(list(self.classifier.tags_binarizer.classes_), [self.t1.pk, self.t3.pk]) - def testPredict(self): self.generate_test_data() self.classifier.train() self.assertEqual(self.classifier.predict_correspondent(self.doc1.content), self.c1.pk) self.assertEqual(self.classifier.predict_correspondent(self.doc2.content), None) self.assertTupleEqual(self.classifier.predict_tags(self.doc1.content), (self.t1.pk,)) - self.assertTupleEqual(self.classifier.predict_tags(self.doc2.content), (self.t1.pk,self.t3.pk)) + self.assertTupleEqual(self.classifier.predict_tags(self.doc2.content), (self.t1.pk, self.t3.pk)) self.assertEqual(self.classifier.predict_document_type(self.doc1.content), self.dt.pk) self.assertEqual(self.classifier.predict_document_type(self.doc2.content), None)