diff --git a/stix2/test/test_granular_markings.py b/stix2/test/test_granular_markings.py index 50fc27d..2f0524b 100644 --- a/stix2/test/test_granular_markings.py +++ b/stix2/test/test_granular_markings.py @@ -1,7 +1,7 @@ import pytest -from stix2 import Malware, exceptions, markings +from stix2 import Malware, markings from .constants import FAKE_TIME, MALWARE_ID, MARKING_IDS from .constants import MALWARE_KWARGS as MALWARE_KWARGS_CONST diff --git a/stix2/test/test_markings.py b/stix2/test/test_markings.py index 726cd98..63aecc3 100644 --- a/stix2/test/test_markings.py +++ b/stix2/test/test_markings.py @@ -155,8 +155,8 @@ def test_campaign_with_granular_markings_example(): marking_ref="marking-definition--613f2e26-407d-48c7-9eca-b8e91df99dc9", selectors=["description"]) ]) - print (markings.get_markings(campaign, None)) - print (markings.add_markings(campaign, None, "marking-definition--00000000-0000-0000-0000-000000000000")) + print(markings.get_markings(campaign, None)) + print(markings.add_markings(campaign, None, "marking-definition--00000000-0000-0000-0000-000000000000")) assert str(campaign) == EXPECTED_CAMPAIGN_WITH_GRANULAR_MARKINGS diff --git a/stix2/test/test_object_markings.py b/stix2/test/test_object_markings.py index 04c1b97..25792a7 100644 --- a/stix2/test/test_object_markings.py +++ b/stix2/test/test_object_markings.py @@ -23,7 +23,7 @@ def test_add_markings_one_marking(): after = Malware( object_marking_refs=[MARKING_IDS[0]], - **MALWARE_KWARGS, + **MALWARE_KWARGS ) before = markings.add_markings(before, None, MARKING_IDS[0]) @@ -39,7 +39,7 @@ def test_add_markings_multiple_marking(): after = Malware( object_marking_refs=[MARKING_IDS[0], MARKING_IDS[1]], - **MALWARE_KWARGS, + **MALWARE_KWARGS ) before = markings.add_markings(before, None, [MARKING_IDS[0], MARKING_IDS[1]]) @@ -62,7 +62,7 @@ def test_add_markings_combination(): { "selectors": ["name"], "marking_ref": MARKING_IDS[3] - }, + } ], **MALWARE_KWARGS )