Browse Source

Refs #373 -- Fixed CompositePrimaryKey tests if yaml isn't installed.

Sarah Boyce 3 tháng trước cách đây
mục cha
commit
28f81a1019
1 tập tin đã thay đổi với 8 bổ sung1 xóa
  1. 8 1
      tests/composite_pk/tests.py

+ 8 - 1
tests/composite_pk/tests.py

@@ -1,7 +1,13 @@
 import json
+import unittest
 from uuid import UUID
 
-import yaml
+try:
+    import yaml  # NOQA
+
+    HAS_YAML = True
+except ImportError:
+    HAS_YAML = False
 
 from django import forms
 from django.core import serializers
@@ -252,6 +258,7 @@ class CompositePKFixturesTests(TestCase):
             },
         )
 
+    @unittest.skipUnless(HAS_YAML, "No yaml library detected")
     def test_serialize_user_yaml(self):
         users = User.objects.filter(pk=(2, 3))
         result = serializers.serialize("yaml", users)