fixed formatting issues

This commit is contained in:
Yannik Henke 2025-03-05 10:56:03 +01:00
parent df46f71980
commit cd47a0867e
2 changed files with 25 additions and 5 deletions

View file

@ -60,7 +60,10 @@ class Requirement:
)
def contains_version_specifier(self) -> bool:
return bool(self.VERSION_SPECIFIED.match(self.value))
return (
self.value is not None and
bool(self.VERSION_SPECIFIED.match(self.value))
)
def append_value(self, value: bytes) -> None:
if self.value is not None:
@ -171,7 +174,9 @@ def main(argv: Sequence[str] | None = None) -> int:
for arg in args.filenames:
with open(arg, 'rb+') as file_obj:
ret_for_file = fix_requirements(file_obj, args.fail_without_version)
ret_for_file = fix_requirements(
file_obj, args.fail_without_version,
)
if ret_for_file:
print(f'Error in file {arg}')

View file

@ -120,9 +120,24 @@ from pre_commit_hooks.requirements_txt_fixer import Requirement
b'a=2.0.0 \\\n --hash=sha256:abcd\nb==1.0.0\n',
),
(b'bar\nfoo\n', ['--fail-without-version'], FAIL, b'bar\nfoo\n'),
(b'bar==1.0\nfoo==1.1a\n', ['--fail-without-version'], PASS, b'bar==1.0\nfoo==1.1a\n'),
(b'#test\nbar==1.0\nfoo==1.1a\n', ['--fail-without-version'], PASS, b'#test\nbar==1.0\nfoo==1.1a\n'),
(b'bar==1.0\n#test\nfoo==1.1a\n', ['--fail-without-version'], PASS, b'bar==1.0\n#test\nfoo==1.1a\n'),
(
b'bar==1.0\nfoo==1.1a\n',
['--fail-without-version'],
PASS,
b'bar==1.0\nfoo==1.1a\n',
),
(
b'#test\nbar==1.0\nfoo==1.1a\n',
['--fail-without-version'],
PASS,
b'#test\nbar==1.0\nfoo==1.1a\n',
),
(
b'bar==1.0\n#test\nfoo==1.1a\n',
['--fail-without-version'],
PASS,
b'bar==1.0\n#test\nfoo==1.1a\n',
),
),
)
def test_integration(input_s, argv, expected_retval, output, tmpdir):