X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Ftest_srcformats.py;h=1e78981b6a9ec45dc3c5cbb4a7bf5b37a0881d02;hb=8c266683b16ad4f1ddcf2cc48cc7931ac007af59;hp=f6d7215fc8daabcd179543ff59691263ccbb9d23;hpb=1ef1804c272cf73da57a3e2d1cbf6c7610207ee9;p=dak.git

diff --git a/tests/test_srcformats.py b/tests/test_srcformats.py
index f6d7215f..1e78981b 100755
--- a/tests/test_srcformats.py
+++ b/tests/test_srcformats.py
@@ -8,6 +8,7 @@ sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from collections import defaultdict
 
 from daklib import srcformats
+from daklib.formats import parse_format
 from daklib.dak_exceptions import UnknownFormatError
 
 class SourceFormatTestCase(unittest.TestCase):
@@ -107,12 +108,12 @@ class FormatTreeQuiltTestCase(SourceFormatTestCase):
 
 class ParseFormatTestCase(unittest.TestCase):
     def assertParse(self, format, expected):
-        self.assertEqual(srcformats.parse_format(format), expected)
+        self.assertEqual(parse_format(format), expected)
 
     def assertParseFail(self, format):
         self.assertRaises(
             UnknownFormatError,
-            lambda: srcformats.parse_format(format)
+            lambda: parse_format(format)
         )
 
     def testParse(self):