Move parse_format into daklib.formats

[dak.git] / tests / test_srcformats.py

diff --git a/tests/test_srcformats.py b/tests/test_srcformats.py
index f6d7215fc8daabcd179543ff59691263ccbb9d23..1e78981b6a9ec45dc3c5cbb4a7bf5b37a0881d02 100755 (executable)
--- 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):