def test_clear_pycondor_environment_variables():
# Set pycondor-related environment variables
for i in ['submit', 'output', 'error', 'log']:
os.environ['PYCONDOR_{}_DIR'.format(i.upper())] = 'something'
# Test that environment variables are cleared
clear_pycondor_environment_variables()
for i in ['submit', 'output', 'error', 'log']:
assert os.getenv('PYCONDOR_{}_DIR'.format(i.upper())) is None
import os
import sys
import shutil
from distutils import spawn
import warnings
import pytest
from pycondor import Job, Dagman
from pycondor.utils import clear_pycondor_environment_variables
warnings.simplefilter('always')
clear_pycondor_environment_variables()
here = os.path.abspath(os.path.dirname(__file__))
example_script = os.path.join(here, 'example_script.py')
@pytest.fixture()
def job(tmpdir_factory):
submit_dir = str(tmpdir_factory.mktemp('submit'))
job = Job('jobname', example_script, submit=submit_dir)
return job
@pytest.fixture()
def monkeypatch_condor_submit(monkeypatch):
# Want to monkeypatch shutil.which to mimic condor_submit existing
version_major = sys.version_info.major
version_minor = sys.version_info.minor
if (version_major, version_minor) >= (3, 3):
monkeypatch.setattr(shutil, 'which', lambda x: 'submit_exists.exe')
else:
def test_clear_pycondor_environment_variables():
# Test that environment variables are cleared
clear_pycondor_environment_variables()
for i in ['submit', 'output', 'error', 'log']:
assert os.environ['PYCONDOR_{}_DIR'.format(i.upper())] == ''