TeamCity python test suite loader/runner

__main__.py

from teamcity import is_running_under_teamcity
from teamcity.unittestpy import TeamcityTestRunner

import os
import unittest

# update these variables to configure how the test runner loads the test suite
testsPath = './tests'
testsPattern = '*_test.py'

def loadSuite():
	return unittest.TestLoader().discover(testsPath, testsPattern)

def createRunner():
	if is_running_under_teamcity():
		print("TeamCity Environment Detected")
		return TeamcityTestRunner()
	else:
		return unittest.TextTestRunner()

# run the suite
createRunner().run(loadSuite())