codestyle

This commit is contained in:
Jonas Winkler
2020-11-18 22:41:14 +01:00
parent 90ead9bf8d
commit 8c40c54421
9 changed files with 8 additions and 19 deletions

View File

@@ -2,11 +2,10 @@ import os
import shutil
import tempfile
from unittest import mock
from unittest.mock import MagicMock
from django.contrib.auth.models import User
from django.test import override_settings
from rest_framework.test import APITestCase, APIClient
from rest_framework.test import APITestCase
from documents.models import Document, Correspondent, DocumentType, Tag

View File

@@ -80,6 +80,6 @@ class TestClassifier(TestCase):
self.classifier.save_classifier()
newClassifier = DocumentClassifier()
newClassifier.reload()
self.assertFalse(newClassifier.train())
new_classifier = DocumentClassifier()
new_classifier.reload()
self.assertFalse(new_classifier.train())

View File

@@ -5,8 +5,6 @@ import tempfile
from unittest import mock
from unittest.mock import MagicMock
from django.conf import settings
from django.db import DatabaseError
from django.test import TestCase, override_settings
from ..consumer import Consumer, ConsumerError
@@ -504,9 +502,9 @@ class TestConsumer(TestCase):
def testOverrideFilename(self):
filename = self.get_test_file()
overrideFilename = "My Bank - Statement for November.pdf"
override_filename = "My Bank - Statement for November.pdf"
document = self.consumer.try_consume_file(filename, override_filename=overrideFilename)
document = self.consumer.try_consume_file(filename, override_filename=override_filename)
self.assertEqual(document.correspondent.name, "My Bank")
self.assertEqual(document.title, "Statement for November")