def test_correct_number_suffixes(
parse_tokens,
assert_errors,
default_options,
code,
):
"""Ensures that correct numbers are fine."""
file_tokens = parse_tokens(code)
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [])
def test_unicode_prefix(
parse_tokens,
assert_errors,
default_options,
code,
):
"""Ensures that unicode prefixes raise a warning."""
file_tokens = parse_tokens(code)
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [UnicodeStringViolation])
def test_bad_number_suffixes(
parse_tokens,
assert_errors,
default_options,
code,
):
"""Ensures that numbers with suffix not in lowercase raise a warning."""
file_tokens = parse_tokens(code)
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [BadNumberSuffixViolation])
def test_underscored_number(
parse_tokens,
assert_errors,
default_options,
code,
):
"""Ensures that underscored numbers raise a warning."""
file_tokens = parse_tokens(code)
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [UnderscoredNumberViolation])
def test_partial_float(
parse_tokens,
assert_errors,
default_options,
code,
):
"""Ensures that partial floats raise a warning."""
file_tokens = parse_tokens(code)
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [PartialFloatViolation])
def test_correct_number(
parse_tokens,
assert_errors,
default_options,
primitives_usages,
primitive,
mode,
):
"""Ensures that correct numbers are fine."""
file_tokens = parse_tokens(mode(primitives_usages.format(primitive)))
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [])
def test_similar_strings(
parse_tokens,
assert_errors,
default_options,
code,
number,
mode,
):
"""Ensures that strings are fine."""
file_tokens = parse_tokens(mode(code.format(number)))
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [])
def test_underscored_number(
parse_tokens,
assert_errors,
assert_error_text,
default_options,
primitives_usages,
primitive,
mode,
):
"""Ensures that underscored numbers raise a warning."""
file_tokens = parse_tokens(mode(primitives_usages.format(primitive)))
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [UnderscoredNumberViolation])
assert_error_text(visitor, primitive.replace('-', ''))
def test_partial_float(
parse_tokens,
assert_errors,
assert_error_text,
default_options,
primitives_usages,
primitive,
mode,
):
"""Ensures that partial floats raise a warning."""
file_tokens = parse_tokens(mode(primitives_usages.format(primitive)))
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [PartialFloatViolation])
assert_error_text(visitor, primitive.replace('-', ''))
def test_bad_number_suffixes(
parse_tokens,
assert_errors,
assert_error_text,
default_options,
primitives_usages,
number,
mode,
):
"""Ensures that numbers with suffix not in lowercase raise a warning."""
file_tokens = parse_tokens(mode(primitives_usages.format(number)))
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [BadNumberSuffixViolation])
assert_error_text(visitor, number.replace('-', ''))
def test_correct_prefix(
parse_tokens,
assert_errors,
default_options,
primitives_usages,
primitive,
modifier,
mode,
):
"""Ensures that correct prefixes work."""
string = primitive.format(modifier)
file_tokens = parse_tokens(mode(primitives_usages.format(string)))
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [])
def test_uppercase_prefix(
parse_tokens,
assert_errors,
assert_error_text,
default_options,
primitives_usages,
primitive,
modifier,
mode,
):
"""Ensures that uppercase modifiers are restricted."""
string = primitive.format(modifier)
file_tokens = parse_tokens(mode(primitives_usages.format(string)))
visitor = WrongPrimitivesVisitor(default_options, file_tokens=file_tokens)
visitor.run()
assert_errors(visitor, [UppercaseStringModifierViolation])
assert_error_text(visitor, modifier)