Advertisement
Guest User

Untitled

a guest
Oct 16th, 2019
117
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.49 KB | None | 0 0
  1. def generate_test_numbers(test_class):
  2.     counter = 1
  3.     for method_name in dir(test_class):
  4.         if not method_name.startswith('test_N_'):
  5.             continue
  6.         method = getattr(test_class, method_name)
  7.         if not callable(method):
  8.             continue
  9.         new_method_name = method_name.replace('_N_', '_{:02d}_'.format(counter))
  10.         counter += 1
  11.         setattr(test_class, new_method_name, method)
  12.         delattr(test_class, method_name)
  13.     return test_class
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement