Add unit test for conda environment management

- Also fix a bug caught by unit test!
This commit is contained in:
yuvipanda
2018-06-26 04:36:20 -07:00
parent 6dc68210c2
commit 92558e913e
2 changed files with 56 additions and 2 deletions

View File

@@ -26,7 +26,7 @@ def ensure_conda_env(prefix):
def ensure_conda_packages(prefix, packages):
"""
Ensure packages are installed in the conda prefix
Ensure packages are installed in the conda prefix.
"""
abspath = os.path.abspath(prefix)
# Let subprocess errors propagate
@@ -41,7 +41,8 @@ def ensure_conda_packages(prefix, packages):
# parse this outside of this kludge.
filtered_output = '\n'.join([
l for l in raw_output.split('\n')
if not l.startswith('{"fetch"')
# Sometimes the JSON messages start with a \00. The lstrip removes these.
if not l.lstrip().startswith('{"fetch"')
])
output = json.loads(filtered_output)
if 'success' in output and output['success'] == True: