add yaml to result

Author: Cody D'Ambrosio

src/tap/line.py

@@ -111,7 +111,12 @@ def __str__(self):
         diagnostics = ""
         if self.diagnostics is not None:
             diagnostics = "\n" + self.diagnostics.rstrip()
-        return f"{is_not}ok {self.number} {self.description}{directive}{diagnostics}"
+        yaml_block = ""
+        if self._yaml_block is not None:
+            indent = " "
+            indented_yaml_block = '\n'.join(f"{indent}{line}" for line in self._yaml_block.splitlines())
+            yaml_block = f"\n{indent}---\n{indented_yaml_block}\n{indent}..."
+        return f"{is_not}ok {self.number} {self.description}{directive}{diagnostics}{yaml_block}"
 
 
 class Plan(Line):

src/tap/tracker.py

@@ -73,14 +73,7 @@ def _track(self, class_name):
             if self.combined:
                 self.combined_test_cases_seen.append(class_name)
 
-    def add_ok(
-        self,
-        class_name,
-        description,
-        directive="",
-        diagnostics=None,
-        raw_yaml_block=None,
-    ):
+    def add_ok(self, class_name, description, directive="", diagnostics=None, raw_yaml_block=None):
         result = Result(
             ok=True,
             number=self._get_next_line_number(class_name),
@@ -91,21 +84,14 @@ def add_ok(
         )
         self._add_line(class_name, result)
 
-    def add_not_ok(
-        self,
-        class_name,
-        description,
-        directive="",
-        diagnostics=None,
-        raw_yaml_block=None,
-    ):
+    def add_not_ok(self, class_name, description, directive="", diagnostics=None, raw_yaml_block=None):
         result = Result(
             ok=False,
             number=self._get_next_line_number(class_name),
             description=description,
             diagnostics=diagnostics,
             directive=Directive(directive),
-            raw_yaml_block=raw_yaml_block,
+            raw_yaml_block=raw_yaml_block
         )
         self._add_line(class_name, result)