changing environment variables

tests/apiclient.py

@@ -46,6 +46,25 @@ class TestAPIClient(unittest.TestCase):
         
         result = self.api_client.run_job(mock_job, "value1")
         self.assertEqual(result, "Job executed with value1")
+
+
+    def test_set_production_env(self):
+        Prod_API = APIClient(
+            client_id="prod_client_id",
+            client_secret="prod_client_secret",
+            debug=False,
+            environment="production"
+        )
+        self.assertEqual(Prod_API.environment, "production")
+
+    def test_set_bad_env(self):
+        with self.assertRaises(ValueError):
+            APIClient(
+                client_id="test_client_id",
+                client_secret="test_client_secret",
+                debug=False,
+                environment="invalid_env"
+            )
         
     @patch('trustcafeapiwrapper.apiclient.requests.request')
     def test_make_request(self, mock_request):