Index: llvm/trunk/test/tools/yaml2obj/lit.local.cfg
===================================================================
--- llvm/trunk/test/tools/yaml2obj/lit.local.cfg
+++ llvm/trunk/test/tools/yaml2obj/lit.local.cfg
@@ -0,0 +1 @@
+config.suffixes = ['.yaml']
Index: llvm/trunk/test/tools/yaml2obj/missing_document_tag.yaml
===================================================================
--- llvm/trunk/test/tools/yaml2obj/missing_document_tag.yaml
+++ llvm/trunk/test/tools/yaml2obj/missing_document_tag.yaml
@@ -4,3 +4,6 @@
 DummyData:
   foo:           0
 ...
+
+# CHECK: YAML:4:1: error: YAML Object File missing document type tag!
+# CHECK: yaml2obj: Failed to parse YAML file!
Index: llvm/trunk/test/tools/yaml2obj/unsupported_document_tag.yaml
===================================================================
--- llvm/trunk/test/tools/yaml2obj/unsupported_document_tag.yaml
+++ llvm/trunk/test/tools/yaml2obj/unsupported_document_tag.yaml
@@ -5,4 +5,4 @@
   foo:           0
 ...
 
-#check error: YAML Object File unsupported document type tag '!unsupported-tag'!
+# CHECK: error: YAML Object File unsupported document type tag '!unsupported-tag'!