I'm using python 3.7 and os library. I have to run a battery of tests on both STAGE and TEST environments. Currently the code sets the environment to STAGE 

> 
> ENVIRONMENT = os.getenv('ENV', 'STAGE')
> 
> 
> 

but I want it to be set by passing an argument via command line. Anyone?