paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-10-30 03:56:23 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix formatting

Browse Source
This commit is contained in:
Daniel Quinn
2018-02-18 18:00:34 +00:00
parent 9e26e7b39e
commit fc6d2d5e0c
1 changed files with 76 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

138
src/paperless_tesseract/tests/test_date.py
View File

@@ -28,9 +28,8 @@ class TestDate(TestCase):
def test_date_format_1(self): def test_date_format_1(self):
input_file = os.path.join(self.SAMPLE_FILES, "") input_file = os.path.join(self.SAMPLE_FILES, "")
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
self.assertEqual(document.get_date(),
None)
document._text = "lorem ipsum 130218 lorem ipsum" document._text = "lorem ipsum 130218 lorem ipsum"
self.assertEqual(document.get_date(), None)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -39,9 +38,8 @@ class TestDate(TestCase):
def test_date_format_2(self): def test_date_format_2(self):
input_file = os.path.join(self.SAMPLE_FILES, "") input_file = os.path.join(self.SAMPLE_FILES, "")
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
self.assertEqual(document.get_date(),
None)
document._text = "lorem ipsum 2018 lorem ipsum" document._text = "lorem ipsum 2018 lorem ipsum"
self.assertEqual(document.get_date(), None)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -50,9 +48,8 @@ class TestDate(TestCase):
def test_date_format_3(self): def test_date_format_3(self):
input_file = os.path.join(self.SAMPLE_FILES, "") input_file = os.path.join(self.SAMPLE_FILES, "")
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
self.assertEqual(document.get_date(),
None)
document._text = "lorem ipsum 20180213 lorem ipsum" document._text = "lorem ipsum 20180213 lorem ipsum"
self.assertEqual(document.get_date(), None)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -61,10 +58,11 @@ class TestDate(TestCase):
def test_date_format_4(self): def test_date_format_4(self):
input_file = os.path.join(self.SAMPLE_FILES, "") input_file = os.path.join(self.SAMPLE_FILES, "")
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
self.assertEqual(document.get_date(),
datetime.datetime(2018, 2, 13, 0, 0,
tzinfo=tz.tzutc()))
document._text = "lorem ipsum 13.02.2018 lorem ipsum" document._text = "lorem ipsum 13.02.2018 lorem ipsum"
self.assertEqual(
document.get_date(),
datetime.datetime(2018, 2, 13, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -73,11 +71,12 @@ class TestDate(TestCase):
def test_date_format_5(self): def test_date_format_5(self):
input_file = os.path.join(self.SAMPLE_FILES, "") input_file = os.path.join(self.SAMPLE_FILES, "")
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
self.assertEqual(document.get_date(),
datetime.datetime(2018, 2, 13, 0, 0,
tzinfo=tz.tzutc()))
document._text = ( document._text = (
"lorem ipsum 130218, 2018, 20180213 and 13.02.2018 lorem ipsum") "lorem ipsum 130218, 2018, 20180213 and 13.02.2018 lorem ipsum")
self.assertEqual(
document.get_date(),
datetime.datetime(2018, 2, 13, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -86,8 +85,6 @@ class TestDate(TestCase):
def test_date_format_6(self): def test_date_format_6(self):
input_file = os.path.join(self.SAMPLE_FILES, "") input_file = os.path.join(self.SAMPLE_FILES, "")
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
self.assertEqual(document.get_date(),
None)
document._text = ( document._text = (
"lorem ipsum\n" "lorem ipsum\n"
"Wohnort\n" "Wohnort\n"
@@ -99,6 +96,7 @@ class TestDate(TestCase):
"BIC\n" "BIC\n"
"lorem ipsum" "lorem ipsum"
) )
self.assertEqual(document.get_date(), None)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -107,14 +105,15 @@ class TestDate(TestCase):
def test_date_format_7(self): def test_date_format_7(self):
input_file = os.path.join(self.SAMPLE_FILES, "") input_file = os.path.join(self.SAMPLE_FILES, "")
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
self.assertEqual(document.get_date(),
datetime.datetime(2019, 3, 1, 0, 0,
tzinfo=tz.tzutc()))
document._text = ( document._text = (
"lorem ipsum\n" "lorem ipsum\n"
"März 2019\n" "März 2019\n"
"lorem ipsum" "lorem ipsum"
) )
self.assertEqual(
document.get_date(),
datetime.datetime(2019, 3, 1, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -125,9 +124,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 4, 1, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 4, 1, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -138,9 +138,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), False) self.assertEqual(document._is_ocred(), False)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 4, 1, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 4, 1, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -151,9 +152,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2013, 2, 1, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2013, 2, 1, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -164,9 +166,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), False) self.assertEqual(document._is_ocred(), False)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2013, 2, 1, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2013, 2, 1, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -177,9 +180,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 10, 5, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 10, 5, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -190,9 +194,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), False) self.assertEqual(document._is_ocred(), False)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 10, 5, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 10, 5, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -203,9 +208,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 10, 5, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 10, 5, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -216,9 +222,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), False) self.assertEqual(document._is_ocred(), False)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 10, 5, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 10, 5, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -229,9 +236,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 12, 17, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 12, 17, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -242,9 +250,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), False) self.assertEqual(document._is_ocred(), False)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 12, 17, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 12, 17, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -256,9 +265,10 @@ class TestDate(TestCase):
document.get_text() document.get_text()
document.DATE_ORDER = "MDY" document.DATE_ORDER = "MDY"
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 12, 17, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 12, 17, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -270,9 +280,10 @@ class TestDate(TestCase):
document.get_text() document.get_text()
document.DATE_ORDER = "MDY" document.DATE_ORDER = "MDY"
self.assertEqual(document._is_ocred(), False) self.assertEqual(document._is_ocred(), False)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 12, 17, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 12, 17, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -305,9 +316,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2018, 4, 1, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2018, 4, 1, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -318,9 +330,10 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2017, 12, 31, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2017, 12, 31, 0, 0, tzinfo=tz.tzutc())
)
@mock.patch( @mock.patch(
"paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH", "paperless_tesseract.parsers.RasterisedDocumentParser.SCRATCH",
@@ -331,6 +344,7 @@ class TestDate(TestCase):
document = RasterisedDocumentParser(input_file) document = RasterisedDocumentParser(input_file)
document.get_text() document.get_text()
self.assertEqual(document._is_ocred(), True) self.assertEqual(document._is_ocred(), True)
self.assertEqual(document.get_date(), self.assertEqual(
datetime.datetime(2017, 12, 31, 0, 0, document.get_date(),
tzinfo=tz.tzutc())) datetime.datetime(2017, 12, 31, 0, 0, tzinfo=tz.tzutc())
)