Improve test coverage

Fixes #12

tests/test_read.py

@@ -1,6 +1,4 @@
-import os
 import pytest
-import sys
 from io import StringIO
 
 from wosfile.read import (
@@ -97,6 +95,12 @@ def test_forgotten_EF(self):
         with pytest.raises(ReadError):
             list(r)
 
+    def test_unknown_tag(self):
+        f = StringIO(preamble_s + "PT abc\n\nQQ x\nER\nEF")
+        r = PlainTextReader(f)
+        with pytest.raises(NotImplementedError):
+            list(r)
+
     def test_ignore_empty_lines(self):
         f = StringIO(preamble_s + "PT abc\n\nAU xyz\nER\nEF")
         r = PlainTextReader(f)

tests/test_record.py

@@ -66,6 +66,11 @@ def test_record_author_address(data, record_id, author_address):
     assert rec.author_address == author_address
 
 
+def test_record_author_address_missing():
+    rec = Record({})
+    assert rec.author_address is None
+
+
 addresses = [
     # (input, expected, exception)
     (

wosfile/read.py

@@ -1,7 +1,6 @@
 import codecs
 import logging
 import pathlib
-import sys
 from csv import DictReader
 from typing import (
     AnyStr,
@@ -217,11 +216,13 @@ def _format_values(self, heading: str, values: List[str]) -> str:
             else:
                 return " ".join(values)
         except KeyError:
-            msg = "\n------------ ERROR ------------\n" \
-                "Seems that the tag \"{}\" is new and not yet handled by the wosfile library.\n" \
-                "Please report this error:\n" \
-                "  https://github.com/rafguns/wosfile/issues\n" \
+            msg = (
+                "\n------------ ERROR ------------\n"
+                'Seems that the tag "{}" is new and not yet handled by the wosfile library.\n'
+                "Please report this error:\n"
+                "  https://github.com/rafguns/wosfile/issues\n"
                 "We are sorry for the inconvenience.\n"
+            )
             raise NotImplementedError(msg.format(heading))
 
     def __next__(self) -> Dict[str, str]:

wosfile/record.py

@@ -22,6 +22,7 @@ def __init__(
         :param bool skip_empty: whether or not to skip empty fields
 
         """
+        super().__init__(self)
         self.skip_empty = skip_empty
         if wos_data:
             self.parse(wos_data)