def test_each_suite_gets_one_test(self):
suites = [grs.Suite(), grs.Suite(), grs.Suite()]
tests_runtimes = self.generate_tests_runtimes(3)
grs.divide_remaining_tests_among_suites(tests_runtimes, suites)
for suite in suites:
self.assertEqual(suite.get_test_count(), 1)
def test_each_suite_gets_at_least_one_test(self):
suites = [grs.Suite(), grs.Suite(), grs.Suite()]
tests_runtimes = self.generate_tests_runtimes(5)
grs.divide_remaining_tests_among_suites(tests_runtimes, suites)
total_tests = 0
for suite in suites:
total_tests += suite.get_test_count()
self.assertGreaterEqual(suite.get_test_count(), 1)
self.assertEqual(total_tests, len(tests_runtimes))
def test_model_generation(self):
suite = grs.Suite()
suite.add_test(
'test1', {
"max_runtime": 10 * 60,
"variant1": 5 * 60,
"variant2": 10 * 60,
"variant3": 7 * 60,
})
suite.add_test(
'test2', {
"max_runtime": 12 * 60,
"variant1": 12 * 60,
"variant2": 8 * 60,
"variant3": 6 * 60,
})
suite.add_test(
'test3', {
"max_runtime": 7 * 60,
"variant1": 6 * 60,
"variant2": 6 * 60,
"variant3": 7 * 60,
})
model = suite.get_model()
self.assertEqual(model["test_names"], ["test1", "test2", "test3"])
self.assertIn({"runtime": 23, "name": "variant1"}, model["variants"])
self.assertIn({"runtime": 24, "name": "variant2"}, model["variants"])
self.assertIn({"runtime": 20, "name": "variant3"}, model["variants"])
def test_adding_tests_increases_count_and_runtime(self):
suite = grs.Suite()
suite.add_test('test1', 10)
suite.add_test('test2', 12)
suite.add_test('test3', 7)
self.assertEqual(suite.get_test_count(), 3)
self.assertEqual(suite.get_runtime(), 29)
def test_adding_tests_increases_count_and_runtime(self):
suite = grs.Suite()
suite.add_test('test1', {
"max_runtime": 10,
"variant1": 5,
"variant2": 10,
"variant3": 7,
})
suite.add_test('test2', {
"max_runtime": 12,
"variant1": 12,
"variant2": 8,
"variant3": 6,
})
suite.add_test('test3', {
"max_runtime": 7,
"variant1": 6,
"variant2": 6,
"variant3": 7,
})
self.assertEqual(suite.get_test_count(), 3)
self.assertEqual(suite.get_runtime(), 29)
def create_suite(count=3, start=0):
""" Create a suite with count tests."""
suite = grs.Suite()
for i in range(start, start + count):
suite.add_test('test{}'.format(i), 1)
return suite
