fix unittests

Author: Cody D'Ambrosio

tests/test_line.py

@@ -3,6 +3,14 @@
 from tap.directive import Directive
 from tap.line import Line, Result
 
+try:
+    import yaml
+    from more_itertools import peekable  # noqa
+
+    have_yaml = True
+except ImportError:
+    have_yaml = False
+
 
 class TestLine(unittest.TestCase):
     """Tests for tap.line.Line"""
@@ -41,11 +49,13 @@ def test_str_diagnostics(self):
         result = Result(False, 45, "failing", diagnostics="# more info")
         self.assertEqual("not ok 45 failing\n# more info", str(result))
 
-    @unittest.mock.patch("tap.line.LOAD_YAML", True)
     def test_yaml_block(self):
         raw_yaml_block = """\
 message: test_message
 severity: fail
 """
         result = Result(False, 46, "passing", None, None, raw_yaml_block=raw_yaml_block)
-        self.assertEqual(result.yaml_block["message"], "test_message")
+        if have_yaml:
+            self.assertEqual(result.yaml_block["message"], "test_message")
+        else:
+            self.assertIsNone(result.yaml_block)

tests/test_tracker.py

@@ -7,6 +7,14 @@
 from tap.tests import TestCase
 from tap.tracker import Tracker
 
+try:
+    import yaml
+    from more_itertools import peekable  # noqa
+
+    have_yaml = True
+except ImportError:
+    have_yaml = False
+
 
 class TestTracker(TestCase):
     def _make_header(self, test_case):
@@ -295,8 +303,6 @@ def test_adds_not_ok_with_diagnostics(self):
         line = tracker._test_cases["FakeTestCase"][0]
         self.assertEqual("# more info\n", line.diagnostics)
 
-    @mock.patch("tap.tracker.ENABLE_VERSION_13", True)
-    @mock.patch("tap.line.LOAD_YAML", True)
     def test_adds_ok_with_yaml_block(self):
         tracker = Tracker()
         tracker.add_ok(
@@ -308,10 +314,11 @@ def test_adds_ok_with_yaml_block(self):
 """,
         )
         line = tracker._test_cases["FakeTestCase"][0]
-        self.assertEqual("test_message", line.yaml_block["message"])
+        if have_yaml:
+            self.assertEqual("test_message", line.yaml_block["message"])
+        else:
+            self.assertIsNone(line.yaml_block)
 
-    @mock.patch("tap.tracker.ENABLE_VERSION_13", True)
-    @mock.patch("tap.line.LOAD_YAML", True)
     def test_adds_not_ok_with_yaml_block(self):
         tracker = Tracker()
         tracker.add_not_ok(
@@ -323,7 +330,10 @@ def test_adds_not_ok_with_yaml_block(self):
 """,
         )
         line = tracker._test_cases["FakeTestCase"][0]
-        self.assertEqual("test_message", line.yaml_block["message"])
+        if have_yaml:
+            self.assertEqual("test_message", line.yaml_block["message"])
+        else:
+            self.assertIsNone(line.yaml_block)
 
     def test_header_displayed_by_default(self):
         tracker = Tracker()