refactor: unittest2pytest -w tests

One step of moving to pure pytest tests.

1 files changed, 6 insertions, 7 deletions

diff --git a/tests/test_xml.py b/tests/test_xml.py
index e3be7a54..033c374e 100644
--- a/tests/test_xml.py
+++ b/tests/test_xml.py
@@ -17,6 +17,7 @@ from coverage.files import abs_file
 
 from tests.coveragetest import CoverageTest
 from tests.goldtest import compare, gold_path
+import pytest
 
 
 class XmlTestHelpers(CoverageTest):
@@ -87,11 +88,11 @@ class XmlTestHelpersTest(XmlTestHelpers, CoverageTest):
         self.assert_source(dom, "something")
         self.assert_source(dom, "another")
 
-        with self.assertRaises(AssertionError):
+        with pytest.raises(AssertionError):
             self.assert_source(dom, "hello")
-        with self.assertRaises(AssertionError):
+        with pytest.raises(AssertionError):
             self.assert_source(dom, "foo")
-        with self.assertRaises(AssertionError):
+        with pytest.raises(AssertionError):
             self.assert_source(dom, "thing")
 
 
@@ -274,10 +275,8 @@ class XmlPackageStructureTest(XmlTestHelpers, CoverageTest):
 
     def assert_package_and_class_tags(self, cov, result):
         """Check the XML package and class tags from `cov` match `result`."""
-        self.assertEqual(
-            unbackslash(list(self.package_and_class_tags(cov))),
-            unbackslash(result),
-            )
+        assert unbackslash(list(self.package_and_class_tags(cov))) == \
+            unbackslash(result)
 
     def test_package_names(self):
         self.make_tree(width=1, depth=3)