test_common_helper.py 939 B

123456789101112131415161718192021222324252627
  1. from facefusion.common_helper import calc_float_step, calc_int_step, create_float_metavar, create_float_range, create_int_metavar, create_int_range
  2. def test_create_int_metavar() -> None:
  3. assert create_int_metavar([ 1, 2, 3, 4, 5 ]) == '[1..5:1]'
  4. def test_create_float_metavar() -> None:
  5. assert create_float_metavar([ 0.1, 0.2, 0.3, 0.4, 0.5 ]) == '[0.1..0.5:0.1]'
  6. def test_create_int_range() -> None:
  7. assert create_int_range(0, 2, 1) == [ 0, 1, 2 ]
  8. assert create_float_range(0, 1, 1) == [ 0, 1 ]
  9. def test_create_float_range() -> None:
  10. assert create_float_range(0.0, 1.0, 0.5) == [ 0.0, 0.5, 1.0 ]
  11. assert create_float_range(0.0, 1.0, 0.05) == [ 0.0, 0.05, 0.10, 0.15, 0.20, 0.25, 0.30, 0.35, 0.40, 0.45, 0.50, 0.55, 0.60, 0.65, 0.70, 0.75, 0.80, 0.85, 0.90, 0.95, 1.0 ]
  12. def test_calc_int_step() -> None:
  13. assert calc_int_step([ 0, 1 ]) == 1
  14. def test_calc_float_step() -> None:
  15. assert calc_float_step([ 0.1, 0.2 ]) == 0.1