pre-commit changes

stix2/__init__.py

@@ -32,40 +32,32 @@ from .confidence import scales
 from .core import _collect_stix2_mappings, parse, parse_observable
 from .datastore import CompositeDataSource
 from .datastore.filesystem import (
-    FileSystemSink, FileSystemSource,
-    FileSystemStore
+    FileSystemSink, FileSystemSource, FileSystemStore,
 )
 from .datastore.filters import Filter
 from .datastore.memory import MemorySink, MemorySource, MemoryStore
 from .datastore.taxii import (
-    TAXIICollectionSink, TAXIICollectionSource,
-    TAXIICollectionStore
+    TAXIICollectionSink, TAXIICollectionSource, TAXIICollectionStore,
 )
 from .environment import Environment, ObjectFactory
 from .markings import (
-    add_markings, clear_markings, get_markings, is_marked,
-    remove_markings, set_markings
+    add_markings, clear_markings, get_markings, is_marked, remove_markings,
+    set_markings,
 )
 from .patterns import (
-    AndBooleanExpression, AndObservationExpression,
-    BasicObjectPathComponent, BinaryConstant,
-    BooleanConstant, EqualityComparisonExpression,
+    AndBooleanExpression, AndObservationExpression, BasicObjectPathComponent,
+    BinaryConstant, BooleanConstant, EqualityComparisonExpression,
     FloatConstant, FollowedByObservationExpression,
-    GreaterThanComparisonExpression,
-    GreaterThanEqualComparisonExpression, HashConstant,
-    HexConstant, InComparisonExpression, IntegerConstant,
-    IsSubsetComparisonExpression,
-    IsSupersetComparisonExpression,
-    LessThanComparisonExpression,
-    LessThanEqualComparisonExpression,
-    LikeComparisonExpression, ListConstant,
-    ListObjectPathComponent, MatchesComparisonExpression,
-    ObjectPath, ObservationExpression, OrBooleanExpression,
-    OrObservationExpression, ParentheticalExpression,
-    QualifiedObservationExpression,
-    ReferenceObjectPathComponent, RepeatQualifier,
-    StartStopQualifier, StringConstant, TimestampConstant,
-    WithinQualifier
+    GreaterThanComparisonExpression, GreaterThanEqualComparisonExpression,
+    HashConstant, HexConstant, InComparisonExpression, IntegerConstant,
+    IsSubsetComparisonExpression, IsSupersetComparisonExpression,
+    LessThanComparisonExpression, LessThanEqualComparisonExpression,
+    LikeComparisonExpression, ListConstant, ListObjectPathComponent,
+    MatchesComparisonExpression, ObjectPath, ObservationExpression,
+    OrBooleanExpression, OrObservationExpression, ParentheticalExpression,
+    QualifiedObservationExpression, ReferenceObjectPathComponent,
+    RepeatQualifier, StartStopQualifier, StringConstant, TimestampConstant,
+    WithinQualifier,
 )
 from .utils import new_version, revoke
 from .v20 import *  # This import will always be the latest STIX 2.X version

stix2/test/v20/test_custom.py

@@ -884,7 +884,8 @@ def test_parse_observable_with_custom_extension():
     assert parsed.extensions['x-new-ext'].property2 == 12
 
 
-@pytest.mark.parametrize("data", [
+@pytest.mark.parametrize(
+    "data", [
         # URL is not in EXT_MAP
         """{
         "type": "url",
@@ -907,7 +908,8 @@ def test_parse_observable_with_custom_extension():
             }
         }
     }""",
-])
+    ],
+)
 def test_parse_observable_with_unregistered_custom_extension(data):
     with pytest.raises(ValueError) as excinfo:
         stix2.parse_observable(data, version='2.0')

stix2/test/v20/test_datastore_memory.py

@@ -289,7 +289,7 @@ def test_memory_store_object_creator_of_present(mem_store):
     iden = Identity(
         id="identity--e4196283-7420-4277-a7a3-d57f61ef1389",
         name="Foo Corp.",
-        identity_class="corporation"
+        identity_class="corporation",
     )
 
     mem_store.add(camp)

stix2/test/v21/test_custom.py

@@ -888,7 +888,8 @@ def test_parse_observable_with_custom_extension():
     assert parsed.extensions['x-new-ext'].property2 == 12
 
 
-@pytest.mark.parametrize("data", [
+@pytest.mark.parametrize(
+    "data", [
         # URL is not in EXT_MAP
         """{
         "type": "url",
@@ -911,7 +912,8 @@ def test_parse_observable_with_custom_extension():
             }
         }
     }""",
-])
+    ],
+)
 def test_parse_observable_with_unregistered_custom_extension(data):
     with pytest.raises(ValueError) as excinfo:
         stix2.parse_observable(data, version='2.1')

stix2/test/v21/test_datastore_memory.py

@@ -10,7 +10,7 @@ from stix2.v21 import (
     Bundle, Campaign, CustomObject, Identity, Indicator, Malware, Relationship,
 )
 
-from stix2.test.v21.constants import (
+from .constants import (
     CAMPAIGN_ID, CAMPAIGN_KWARGS, IDENTITY_ID, IDENTITY_KWARGS, INDICATOR_ID,
     INDICATOR_KWARGS, MALWARE_ID, MALWARE_KWARGS, RELATIONSHIP_IDS,
 )
@@ -296,7 +296,7 @@ def test_memory_store_object_creator_of_present(mem_store):
     iden = Identity(
         id="identity--e4196283-7420-4277-a7a3-d57f61ef1389",
         name="Foo Corp.",
-        identity_class="corporation"
+        identity_class="corporation",
     )
 
     mem_store.add(camp)