Add location tests, fix locations for multi-line items

1 files changed, 9 insertions, 1 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
index aed3dc6..5616c23 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,6 +1,6 @@
 import unittest
 import os
-from tests import parsing, errors
+from tests import parsing, errors, location
 
 class TestParsing(unittest.TestCase):
 	def test_all(self) -> None:
@@ -18,6 +18,14 @@ class TestErrors(unittest.TestCase):
 			with self.subTest(file):
 				errors.test_path(self, f'{test_dir}/{file}')
 
+class TestLocation(unittest.TestCase):
+	def test_all(self) -> None:
+		test_dir = '../tests/location'
+		for file in os.listdir(test_dir):
+			if not file.endswith('.locations.pom'): continue
+			with self.subTest(file):
+				location.test_path(self, f'{test_dir}/{file}')
+
 if __name__ == '__main__':
 	unittest.main()