Come posso riportare la directory alla mia directory di lavoro originale con Python?

Ho una funzione simile a quella qui sotto. Non sono sicuro di come utilizzare il modulo os per tornare alla mia directory di lavoro originale al termine dell'esecuzione del jar.

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

nota: penso che la formattazione del mio codice sia disattivata, non so perché. Mi scuso in anticipo