Как мне вернуться в исходный рабочий каталог с помощью Python?

Вам просто нужно добавить строку:

os.chdir(owd)

Просто примечание, на это также ответили в другом вашем вопрос.

os.chdir (owd) должен помочь (как вы это сделали при переходе на testDir)

Совет использовать os.chdir(owd) хороший. Было бы разумно поместить код, которому требуется измененный каталог, в блок try:finally (или в python 2.6 и более поздних версиях, блок with:). Это снижает риск того, что вы случайно вставите return в код перед тем, как вернуться к исходный каталог.

def run(): 
    owd = os.getcwd()
    try:
        #first change dir to build_dir path
        os.chdir(testDir)
        #run jar from test directory
        os.system(cmd)
    finally:
        #change dir back to original working directory (owd)
        os.chdir(owd)

Python чувствителен к регистру, поэтому при вводе пути убедитесь, что он совпадает с каталогом вы хотите установить.

import os

os.getcwd()

os.chdir('C:\\')

Контекстный менеджер - очень подходящий инструмент для этой работы:

from contextlib import contextmanager

@contextmanager
def cwd(path):
    oldpwd=os.getcwd()
    os.chdir(path)
    try:
        yield
    finally:
        os.chdir(oldpwd)

...используется как:

os.chdir('/tmp') # for testing purposes, be in a known directory
print 'before context manager: %s' % os.getcwd()
with cwd('/'):
    # code inside this block, and only inside this block, is in the new directory
    print 'inside context manager: %s' % os.getcwd()
print 'after context manager: %s' % os.getcwd()

... что даст что-то вроде:

before context manager: /tmp
inside context manager: /
after context manager: /tmp

На самом деле это начальство для встроенной оболочки cd -, поскольку она также заботится об обратном изменении каталогов при выходе из блока из-за генерируемого исключения.

Для вашего конкретного случая использования это будет:

with cwd(testDir):
    os.system(cmd)

Другой вариант, который следует рассмотреть, - использовать subprocess.call() вместо os.system(), что позволит вам указать рабочий каталог для запуска команды:

# note: better to modify this to not need shell=True if possible
subprocess.call(cmd, cwd=testDir, shell=True)

... что избавит вас от необходимости вообще менять каталог интерпретатора.

Контекстного менеджера для этой ситуации (выполнение системной команды) излишне. Лучшее решение - использовать вместо него модуль subprocess (начиная с Python 2.4) и методы run или popen с аргументом cwd.

Итак, ваш код можно заменить на:

def run(): 
    #run jar from test directory
    subprocess.run(cmd, cwd=testDir)

См. https://bugs.python.org/issue25625 и https://docs.python.org/3/library/subprocess.html#subprocess-replacements.