pre-commit styling changes

pull/1/head
Michael Chisholm 2022-04-28 21:23:05 -04:00
parent e534e41865
commit 724122db65
5 changed files with 25 additions and 27 deletions

View File

@ -392,6 +392,6 @@ def create_pattern_object(pattern, module_suffix="", module_name="", version=DEF
pattern_obj = pattern_class(pattern) pattern_obj = pattern_class(pattern)
builder = visitor_class( builder = visitor_class(
parser_class, module_suffix, module_name parser_class, module_suffix, module_name,
) )
return pattern_obj.visit(builder) return pattern_obj.visit(builder)

View File

@ -2,9 +2,7 @@
AST node class overrides for testing the pattern AST builder. AST node class overrides for testing the pattern AST builder.
""" """
from stix2.patterns import ( from stix2.patterns import (
EqualityComparisonExpression, EqualityComparisonExpression, StartStopQualifier, StringConstant,
StartStopQualifier,
StringConstant
) )

View File

@ -5,8 +5,9 @@ import pytz
import stix2 import stix2
from stix2.pattern_visitor import create_pattern_object from stix2.pattern_visitor import create_pattern_object
import stix2.utils
import stix2.patterns import stix2.patterns
import stix2.utils
from .pattern_ast_overrides import * from .pattern_ast_overrides import *
@ -594,7 +595,7 @@ def test_list_constant():
def test_ast_class_override_comp_equals(): def test_ast_class_override_comp_equals():
patt_ast = create_pattern_object( patt_ast = create_pattern_object(
"[a:b=1]", "Testing", "stix2.test.v20.pattern_ast_overrides", "[a:b=1]", "Testing", "stix2.test.v20.pattern_ast_overrides",
version="2.0" version="2.0",
) )
assert isinstance(patt_ast, stix2.patterns.ObservationExpression) assert isinstance(patt_ast, stix2.patterns.ObservationExpression)
@ -606,20 +607,20 @@ def test_ast_class_override_string_constant():
patt_ast = create_pattern_object( patt_ast = create_pattern_object(
"[a:'b'[1].'c' < 'foo']", "Testing", "[a:'b'[1].'c' < 'foo']", "Testing",
"stix2.test.v20.pattern_ast_overrides", "stix2.test.v20.pattern_ast_overrides",
version="2.0" version="2.0",
) )
assert isinstance(patt_ast, stix2.patterns.ObservationExpression) assert isinstance(patt_ast, stix2.patterns.ObservationExpression)
assert isinstance( assert isinstance(
patt_ast.operand, stix2.patterns.LessThanComparisonExpression patt_ast.operand, stix2.patterns.LessThanComparisonExpression,
) )
assert isinstance( assert isinstance(
patt_ast.operand.lhs.property_path[0].property_name, patt_ast.operand.lhs.property_path[0].property_name,
str str,
) )
assert isinstance( assert isinstance(
patt_ast.operand.lhs.property_path[1].property_name, patt_ast.operand.lhs.property_path[1].property_name,
str str,
) )
assert isinstance(patt_ast.operand.rhs, StringConstantForTesting) assert isinstance(patt_ast.operand.rhs, StringConstantForTesting)
@ -629,18 +630,18 @@ def test_ast_class_override_string_constant():
def test_ast_class_override_startstop_qualifier(): def test_ast_class_override_startstop_qualifier():
patt_ast = create_pattern_object( patt_ast = create_pattern_object(
"[a:b=1] START '1993-01-20T01:33:52.592Z' STOP '2001-08-19T23:50:23.129Z'", "[a:b=1] START '1993-01-20T01:33:52.592Z' STOP '2001-08-19T23:50:23.129Z'",
"Testing", "stix2.test.v20.pattern_ast_overrides", version="2.0" "Testing", "stix2.test.v20.pattern_ast_overrides", version="2.0",
) )
assert isinstance(patt_ast, stix2.patterns.QualifiedObservationExpression) assert isinstance(patt_ast, stix2.patterns.QualifiedObservationExpression)
assert isinstance( assert isinstance(
patt_ast.observation_expression, stix2.patterns.ObservationExpression patt_ast.observation_expression, stix2.patterns.ObservationExpression,
) )
assert isinstance( assert isinstance(
patt_ast.observation_expression.operand, patt_ast.observation_expression.operand,
EqualityComparisonExpressionForTesting EqualityComparisonExpressionForTesting,
) )
assert isinstance( assert isinstance(
patt_ast.qualifier, StartStopQualifierForTesting patt_ast.qualifier, StartStopQualifierForTesting,
) )
assert str(patt_ast) == "[a:b = 1] START '1993-01-20T01:33:52.592Z' STOP '2001-08-19T23:50:23.129Z'" assert str(patt_ast) == "[a:b = 1] START '1993-01-20T01:33:52.592Z' STOP '2001-08-19T23:50:23.129Z'"

View File

@ -2,9 +2,7 @@
AST node class overrides for testing the pattern AST builder. AST node class overrides for testing the pattern AST builder.
""" """
from stix2.patterns import ( from stix2.patterns import (
EqualityComparisonExpression, EqualityComparisonExpression, StartStopQualifier, StringConstant,
StartStopQualifier,
StringConstant
) )

View File

@ -6,8 +6,9 @@ from stix2patterns.exceptions import ParseException
import stix2 import stix2
from stix2.pattern_visitor import create_pattern_object from stix2.pattern_visitor import create_pattern_object
import stix2.utils
import stix2.patterns import stix2.patterns
import stix2.utils
from .pattern_ast_overrides import * from .pattern_ast_overrides import *
@ -754,7 +755,7 @@ def test_parse_error():
def test_ast_class_override_comp_equals(): def test_ast_class_override_comp_equals():
patt_ast = create_pattern_object( patt_ast = create_pattern_object(
"[a:b=1]", "Testing", "stix2.test.v21.pattern_ast_overrides", "[a:b=1]", "Testing", "stix2.test.v21.pattern_ast_overrides",
version="2.1" version="2.1",
) )
assert isinstance(patt_ast, stix2.patterns.ObservationExpression) assert isinstance(patt_ast, stix2.patterns.ObservationExpression)
@ -766,20 +767,20 @@ def test_ast_class_override_string_constant():
patt_ast = create_pattern_object( patt_ast = create_pattern_object(
"[a:'b'[1].'c' < 'foo']", "Testing", "[a:'b'[1].'c' < 'foo']", "Testing",
"stix2.test.v21.pattern_ast_overrides", "stix2.test.v21.pattern_ast_overrides",
version="2.1" version="2.1",
) )
assert isinstance(patt_ast, stix2.patterns.ObservationExpression) assert isinstance(patt_ast, stix2.patterns.ObservationExpression)
assert isinstance( assert isinstance(
patt_ast.operand, stix2.patterns.LessThanComparisonExpression patt_ast.operand, stix2.patterns.LessThanComparisonExpression,
) )
assert isinstance( assert isinstance(
patt_ast.operand.lhs.property_path[0].property_name, patt_ast.operand.lhs.property_path[0].property_name,
str str,
) )
assert isinstance( assert isinstance(
patt_ast.operand.lhs.property_path[1].property_name, patt_ast.operand.lhs.property_path[1].property_name,
str str,
) )
assert isinstance(patt_ast.operand.rhs, StringConstantForTesting) assert isinstance(patt_ast.operand.rhs, StringConstantForTesting)
@ -789,18 +790,18 @@ def test_ast_class_override_string_constant():
def test_ast_class_override_startstop_qualifier(): def test_ast_class_override_startstop_qualifier():
patt_ast = create_pattern_object( patt_ast = create_pattern_object(
"[a:b=1] START t'1993-01-20T01:33:52.592Z' STOP t'2001-08-19T23:50:23.129Z'", "[a:b=1] START t'1993-01-20T01:33:52.592Z' STOP t'2001-08-19T23:50:23.129Z'",
"Testing", "stix2.test.v21.pattern_ast_overrides", version="2.1" "Testing", "stix2.test.v21.pattern_ast_overrides", version="2.1",
) )
assert isinstance(patt_ast, stix2.patterns.QualifiedObservationExpression) assert isinstance(patt_ast, stix2.patterns.QualifiedObservationExpression)
assert isinstance( assert isinstance(
patt_ast.observation_expression, stix2.patterns.ObservationExpression patt_ast.observation_expression, stix2.patterns.ObservationExpression,
) )
assert isinstance( assert isinstance(
patt_ast.observation_expression.operand, patt_ast.observation_expression.operand,
EqualityComparisonExpressionForTesting EqualityComparisonExpressionForTesting,
) )
assert isinstance( assert isinstance(
patt_ast.qualifier, StartStopQualifierForTesting patt_ast.qualifier, StartStopQualifierForTesting,
) )
assert str(patt_ast) == "[a:b = 1] START t'1993-01-20T01:33:52.592Z' STOP t'2001-08-19T23:50:23.129Z'" assert str(patt_ast) == "[a:b = 1] START t'1993-01-20T01:33:52.592Z' STOP t'2001-08-19T23:50:23.129Z'"