[test/parser] Update InvalidVersion tests

Author: mpsonntag

test/test_parser_json.py

@@ -3,7 +3,7 @@
 import unittest
 
 from odml.tools import dict_parser
-from odml.tools.parser_utils import ParserException
+from odml.tools.parser_utils import ParserException, InvalidVersionException
 
 
 class TestJSONParser(unittest.TestCase):
@@ -40,12 +40,9 @@ def test_missing_version(self):
 
     def test_invalid_version(self):
         filename = "invalid_version.json"
-        message = "invalid odML document format version"
 
         with open(os.path.join(self.basepath, filename)) as json_data:
             parsed_doc = json.load(json_data)
 
-        with self.assertRaises(ParserException) as exc:
+        with self.assertRaises(InvalidVersionException):
             _ = self.json_reader.to_odml(parsed_doc)
-
-        self.assertIn(message, str(exc.exception))

test/test_parser_xml.py

@@ -2,7 +2,7 @@
 import unittest
 
 from odml.tools import xmlparser
-from odml.tools.parser_utils import ParserException
+from odml.tools.parser_utils import ParserException, InvalidVersionException
 
 
 class TestXMLParser(unittest.TestCase):
@@ -33,9 +33,6 @@ def test_missing_version(self):
 
     def test_invalid_version(self):
         filename = "invalid_version.xml"
-        message = "invalid odML document format version"
 
-        with self.assertRaises(ParserException) as exc:
+        with self.assertRaises(InvalidVersionException):
             _ = self.xml_reader.from_file(os.path.join(self.basepath, filename))
-
-        self.assertIn(message, str(exc.exception))

test/test_parser_yaml.py

@@ -3,7 +3,7 @@
 import yaml
 
 from odml.tools import dict_parser
-from odml.tools.parser_utils import ParserException
+from odml.tools.parser_utils import ParserException, InvalidVersionException
 
 
 class TestYAMLParser(unittest.TestCase):
@@ -40,12 +40,9 @@ def test_missing_version(self):
 
     def test_invalid_version(self):
         filename = "invalid_version.yaml"
-        message = "invalid odML document format version"
 
         with open(os.path.join(self.basepath, filename)) as raw_data:
             parsed_doc = yaml.load(raw_data)
 
-        with self.assertRaises(ParserException) as exc:
+        with self.assertRaises(InvalidVersionException):
             _ = self.yaml_reader.to_odml(parsed_doc)
-
-        self.assertIn(message, str(exc.exception))