update tests

Author: codereverser

casparser/regex.py

@@ -12,7 +12,6 @@
 NAV_RE = r"NAV\s+on\s+(\d{2}-[A-Za-z]{3}-\d{4})\s*:\s*INR\s*([\d,.]+)"
 
 TRANSACTION_RE = (
-    # r"(\d{2}-[A-Za-z]{3}-\d{4})\t\t([^\t]+?)\t\t([(\d,.]+)\)*\t\t([(\d,.]+)\)*\t\t([(\d,.]+)\)*\t\t([(\d,.]+)\)*"
     r"(\d{2}-[A-Za-z]{3}-\d{4})\t\t([^\t]+?)\t\t([(\d,.]+)\)*(?:\t\t([(\d,.]+)\)*\t\t([(\d,.]+)\)*\t\t([(\d,.]+)\)*)*"
 )
 DIVIDEND_RE = r"dividend.+?(reinvest)*.+?@\s+Rs\.\s*([\d\.]+)\s+per\s+unit"

tests/base.py

@@ -24,8 +24,10 @@ def read_pdf(self, filename, password, output="dict"):
         return read_cas_pdf(filename, password, output=output, force_pdfminer=use_pdfminer)
 
     def test_read_dict(self):
-        self.read_pdf(self.cams_file_name, self.cams_password)
-        self.read_pdf(self.kfintech_file_name, self.kfintech_password)
+        data = self.read_pdf(self.cams_file_name, self.cams_password)
+        assert len(data.get('folios', [])) == 10
+        data = self.read_pdf(self.kfintech_file_name, self.kfintech_password)
+        assert len(data.get('folios', [])) == 10
 
     def test_read_json(self):
         self.read_pdf(self.cams_file_name, self.cams_password, output="json")

tests/test_pdfminer.py

@@ -1,5 +1,8 @@
 from click.testing import CliRunner
+from pdfminer.layout import LTTextBoxHorizontal
+import pytest
 
+from casparser.exceptions import CASParseError
 from .base import BaseTestClass
 
 
@@ -20,3 +23,12 @@ def test_cli(self):
         )
         assert result.exit_code == 0
         assert "Statement Period:" in result.output
+
+    def test_bad_investor_info(self):
+        from casparser.parsers.pdfminer import parse_investor_info
+
+        with pytest.raises(CASParseError) as exc_info:
+            box = LTTextBoxHorizontal()
+            box.get_text()
+            parse_investor_info([], 0, 0)
+        assert "Unable to parse investor data" in str(exc_info)

tests/test_process.py

@@ -0,0 +1,17 @@
+import pytest
+
+from casparser.exceptions import HeaderParseError
+from casparser.process import parse_header
+
+
+class TestProcessClass:
+
+    def test_header_parser(self):
+        good_header = 'Consolidated Account Statement\n01-Apr-2018 To 31-Mar-2019'
+        bad_header = 'Consolidated Account Statement\n01-Apr-2018'
+
+        header_data = parse_header(good_header)
+        assert header_data == {"from": "01-Apr-2018", "to": "31-Mar-2019"}
+
+        with pytest.raises(HeaderParseError):
+            parse_header(bad_header)