Python Script to shard UITests in Xcode project

ui_test_sharder.py

import os
import errno
import sys

shard_index = sys.argv[1]

PATH = os.path.dirname(os.getcwd())
TESTSPATH = os.path.join(PATH,"Example/UITests/")
tests = []
shard_tests = ""
target = "Example_UITests"

for _, _, files in os.walk(MODELSPATH):
    for filename in files:
        test_name = os.path.splitext(filename)[0]
        if test_name.endswith('UITests'):
            tests.append(test_name)
tests.sort()            
i = int(shard_index)
while i < len(tests):
  shard_tests += " '-only-testing:%s/%s'" %(target, tests[i])
  i += 2
print shard_tests