def test_if_each_test_should_be_own_suite(self):
max_time = 5
tests_runtimes = [
TestRuntime("test1", 5),
TestRuntime("test2", 4),
TestRuntime("test3", 3),
]
suites = under_test.greedy_division(tests_runtimes, max_time)
self.assertEqual(len(suites), 3)
def test_max_tests_per_suites_is_less_than_number_of_tests(self):
max_time = 100
num_tests = 10
tests_runtimes = [
TestRuntime(f"tests_{i}", 1) for i in range(num_tests)
]
suites = under_test.greedy_division(tests_runtimes,
max_time,
max_tests_per_suite=2)
self.assertEqual(len(suites), num_tests // 2)
def test_max_tests_per_suites_is_one(self):
max_time = 5
num_tests = 10
tests_runtimes = [
TestRuntime(f"tests_{i}", i) for i in range(num_tests)
]
suites = under_test.greedy_division(tests_runtimes,
max_time,
max_tests_per_suite=1)
self.assertEqual(len(suites), num_tests)
def test_if_test_is_greater_than_max_it_goes_alone(self):
max_time = 7
tests_runtimes = [
TestRuntime("test1", 15),
TestRuntime("test2", 4),
TestRuntime("test3", 3),
]
suites = under_test.greedy_division(tests_runtimes, max_time)
self.assertEqual(len(suites), 2)
self.assertEqual(len(suites[0]), 1)
self.assertIn("test1", suites[0])
def test_if_less_total_than_max_only_one_suite_created(self):
max_time = 20
tests_runtimes = [
TestRuntime("test1", 5),
TestRuntime("test2", 4),
TestRuntime("test3", 3),
]
suites = under_test.greedy_division(tests_runtimes, max_time)
self.assertEqual(len(suites), 1)
for test in tests_runtimes:
self.assertIn(test.test_name, suites[0])
def test_max_suites_overrides_max_tests_per_suite(self):
max_time = 100
num_tests = 10
max_suites = 2
tests_runtimes = [
TestRuntime(f"tests_{i}", 1) for i in range(num_tests)
]
suites = under_test.greedy_division(tests_runtimes,
max_time,
max_suites=max_suites,
max_tests_per_suite=2)
self.assertEqual(len(suites), max_suites)
def test_max_sub_suites_options(self):
max_time = 5
max_suites = 2
tests_runtimes = [
TestRuntime("test1", 5),
TestRuntime("test2", 4),
TestRuntime("test3", 3),
TestRuntime("test4", 4),
TestRuntime("test5", 3),
]
suites = under_test.greedy_division(tests_runtimes,
max_time,
max_suites=max_suites)
self.assertEqual(len(suites), max_suites)
total_tests = sum(len(suite) for suite in suites)
self.assertEqual(total_tests, len(tests_runtimes))