Fix style issues

master
Chris Lenk 2019-02-06 15:23:00 -05:00
parent 0ca2bc087a
commit 69d2529f0e
4 changed files with 23 additions and 23 deletions

View File

@ -39,7 +39,7 @@ def _validate_selector(obj, selector):
def _get_marking_id(marking):
if type(marking).__name__ is 'MarkingDefinition': # avoid circular import
if type(marking).__name__ == 'MarkingDefinition': # avoid circular import
return marking.id
return marking

View File

@ -143,7 +143,7 @@ class ListProperty(Property):
if type(self.contained) is EmbeddedObjectProperty:
obj_type = self.contained.type
elif type(self.contained).__name__ is "STIXObjectProperty":
elif type(self.contained).__name__ == "STIXObjectProperty":
# ^ this way of checking doesn't require a circular import
# valid is already an instance of a python-stix2 class; no need
# to turn it into a dictionary and then pass it to the class

View File

@ -1095,11 +1095,11 @@ def test_process_example_empty_error():
def test_process_example_empty_with_extensions():
with pytest.raises(stix2.exceptions.AtLeastOnePropertyError) as excinfo:
stix2.v20.Process(
extensions={
"windows-process-ext": {},
},
)
stix2.v20.Process(
extensions={
"windows-process-ext": {},
},
)
assert excinfo.value.cls == stix2.v20.WindowsProcessExt
properties_of_extension = list(stix2.v20.WindowsProcessExt._properties.keys())
@ -1127,13 +1127,13 @@ def test_process_example_windows_process_ext():
def test_process_example_windows_process_ext_empty():
with pytest.raises(stix2.exceptions.AtLeastOnePropertyError) as excinfo:
stix2.v20.Process(
pid=1221,
name="gedit-bin",
extensions={
"windows-process-ext": {},
},
)
stix2.v20.Process(
pid=1221,
name="gedit-bin",
extensions={
"windows-process-ext": {},
},
)
assert excinfo.value.cls == stix2.v20.WindowsProcessExt
properties_of_extension = list(stix2.v20.WindowsProcessExt._properties.keys())

View File

@ -1073,9 +1073,9 @@ def test_process_example_empty_error():
def test_process_example_empty_with_extensions():
with pytest.raises(stix2.exceptions.AtLeastOnePropertyError) as excinfo:
stix2.v21.Process(extensions={
"windows-process-ext": {},
})
stix2.v21.Process(extensions={
"windows-process-ext": {},
})
assert excinfo.value.cls == stix2.v21.WindowsProcessExt
properties_of_extension = list(stix2.v21.WindowsProcessExt._properties.keys())
@ -1102,12 +1102,12 @@ def test_process_example_windows_process_ext():
def test_process_example_windows_process_ext_empty():
with pytest.raises(stix2.exceptions.AtLeastOnePropertyError) as excinfo:
stix2.v21.Process(
pid=1221,
extensions={
"windows-process-ext": {},
},
)
stix2.v21.Process(
pid=1221,
extensions={
"windows-process-ext": {},
},
)
assert excinfo.value.cls == stix2.v21.WindowsProcessExt
properties_of_extension = list(stix2.v21.WindowsProcessExt._properties.keys())