X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Ftest_srcformats.py;h=9c62d831bdb8a46b609e826848472fca584004f1;hb=c9ba1a13b41a4f21b72485d9d93008f4d16f71ac;hp=802f0ffb688e29ad223760e569213c68c5b3dbe7;hpb=1b29fb2dc99c4569a9aea6a8652ddecc340730d7;p=dak.git

diff --git a/tests/test_srcformats.py b/tests/test_srcformats.py
index 802f0ffb..9c62d831 100755
--- a/tests/test_srcformats.py
+++ b/tests/test_srcformats.py
@@ -107,16 +107,16 @@ class FormatTreeQuiltTestCase(SourceFormatTestCase):
 
 class ParseFormat(unittest.TestCase):
     def assertFormat(self, input, expected, **kwargs):
-        self.assertEqual(
-            srcformats.SourceFormat.parse_format(input, **kwargs),
-            expected,
-        )
+        format = srcformats.SourceFormat.parse_format(input)
+        self.assertEqual(format, expected)
+        srcformats.SourceFormat.validate_format(format, **kwargs)
 
     def assertInvalidFormat(self, input, **kwargs):
-        self.assertRaises(
-            UnknownFormatError,
-            lambda: srcformats.SourceFormat.parse_format(input, **kwargs),
-        )
+        try:
+            format = srcformats.SourceFormat.parse_format(input)
+            srcformats.SourceFormat.validate_format(format, **kwargs)
+        except UnknownFormatError:
+            return
 
     def testEmpty(self):
         self.assertInvalidFormat('')