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)