def setUp(self):
def split(test_name):
idx = test_name.rfind('/')
if idx != -1:
return (test_name[0:idx], test_name[idx + 1:])
self.sharder = Sharder(split, '/', 1)
def get_shards(self, num_workers, fully_parallel, test_list=None):
port = TestPort(MockSystemHost())
self.sharder = Sharder(port.split_test)
test_list = test_list or self.test_list
return self.sharder.shard_tests(
[self.get_test_input(test) for test in test_list], num_workers,
fully_parallel)
def get_shards(self,
num_workers,
fully_parallel,
test_list=None,
max_locked_shards=1):
def split(test_name):
idx = test_name.rfind('/')
if idx != -1:
return (test_name[0:idx], test_name[idx + 1:])
self.sharder = Sharder(split, '/', max_locked_shards)
test_list = test_list or self.test_list
return self.sharder.shard_tests(
[self.get_test_input(test) for test in test_list], num_workers,
fully_parallel)
