e42f91dadf
* Replace audio whenever set via source * use scale_face_landmark_5() in age_modifier * Fix wording and ordering of options * Adjust wording for face editor * Fix wording for processors * Switch order of frame colorizer options * That condition is actual not needed * Simplify UI layout API by removing pre_render() * Clean args and safe cast ini values (#775) * Clean args and safe cast ini values * Clean args and safe cast ini values * Clean args and safe cast ini values * Introduce paths group * Fix job list command and change order * Add job list testing todo * Fix spacing in typing * Fix benchmark by ignoring audio * Simplify and avoid knowing the provider values (#782) * Fix logger table with empty value * Complete Typing --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from argparse import ArgumentParser
|
|
|
|
import pytest
|
|
|
|
from facefusion.program_helper import find_argument_group, remove_args, validate_actions
|
|
|
|
|
|
def test_find_argument_group() -> None:
|
|
program = ArgumentParser()
|
|
program.add_argument_group('test-1')
|
|
program.add_argument_group('test-2')
|
|
|
|
assert find_argument_group(program, 'test-1')
|
|
assert find_argument_group(program, 'test-2')
|
|
assert find_argument_group(program, 'invalid') is None
|
|
|
|
|
|
@pytest.mark.skip()
|
|
def test_validate_args() -> None:
|
|
pass
|
|
|
|
|
|
def test_validate_actions() -> None:
|
|
program = ArgumentParser()
|
|
program.add_argument('--test-1', default = 'test_1', choices = [ 'test_1', 'test_2' ])
|
|
program.add_argument('--test-2', default = 'test_2', choices= [ 'test_1', 'test_2' ], nargs = '+')
|
|
|
|
assert validate_actions(program) is True
|
|
|
|
args =\
|
|
{
|
|
'test_1': 'test_2',
|
|
'test_2': [ 'test_1', 'test_3' ]
|
|
}
|
|
|
|
for action in program._actions:
|
|
if action.dest in args:
|
|
action.default = args[action.dest]
|
|
|
|
assert validate_actions(program) is False
|
|
|
|
|
|
def test_remove_args() -> None:
|
|
program = ArgumentParser()
|
|
program.add_argument('--test-1')
|
|
program.add_argument('--test-2')
|
|
program.add_argument('--test-3')
|
|
|
|
actions = [ action.dest for action in program._actions ]
|
|
|
|
assert 'test_1' in actions
|
|
assert 'test_2' in actions
|
|
assert 'test_3' in actions
|
|
|
|
program = remove_args(program, [ 'test_1', 'test_2' ])
|
|
actions = [ action.dest for action in program._actions ]
|
|
|
|
assert 'test_1' not in actions
|
|
assert 'test_2' not in actions
|
|
assert 'test_3' in actions
|