formatting

Author: dmarticus

posthog/test/test_client.py

@@ -385,20 +385,19 @@ def test_basic_capture_with_locally_evaluated_feature_flags(self, patch_decide):
         assert "$feature/false-flag" not in msg["properties"]
         assert "$active_feature_flags" not in msg["properties"]
 
-
     @mock.patch("posthog.client.get")
     def test_load_feature_flags_quota_limited(self, patch_get):
         mock_response = {
             "type": "quota_limited",
             "detail": "You have exceeded your feature flag request quota",
-            "code": "payment_required"
+            "code": "payment_required",
         }
         patch_get.side_effect = APIError(402, mock_response["detail"])
-        
+
         client = Client(FAKE_TEST_API_KEY, personal_api_key="test")
         with self.assertLogs("posthog", level="WARNING") as logs:
             client._load_feature_flags()
-            
+
             self.assertEqual(client.feature_flags, [])
             self.assertEqual(client.feature_flags_by_key, {})
             self.assertEqual(client.group_type_mapping, {})

posthog/test/test_request.py

@@ -48,30 +48,30 @@ def test_should_timeout(self):
     def test_quota_limited_response(self):
         mock_response = requests.Response()
         mock_response.status_code = 200
-        mock_response._content = json.dumps({
-            "quotaLimited": ["feature_flags"],
-            "featureFlags": {},
-            "featureFlagPayloads": {},
-            "errorsWhileComputingFlags": False
-        }).encode('utf-8')
-
-        with mock.patch('posthog.request._session.post', return_value=mock_response):
+        mock_response._content = json.dumps(
+            {
+                "quotaLimited": ["feature_flags"],
+                "featureFlags": {},
+                "featureFlagPayloads": {},
+                "errorsWhileComputingFlags": False,
+            }
+        ).encode("utf-8")
+
+        with mock.patch("posthog.request._session.post", return_value=mock_response):
             with self.assertRaises(QuotaLimitError) as cm:
                 decide("fake_key", "fake_host")
-            
+
             self.assertEqual(cm.exception.status, 200)
             self.assertEqual(cm.exception.message, "Feature flags quota limited")
 
     def test_normal_decide_response(self):
         mock_response = requests.Response()
         mock_response.status_code = 200
-        mock_response._content = json.dumps({
-            "featureFlags": {"flag1": True},
-            "featureFlagPayloads": {},
-            "errorsWhileComputingFlags": False
-        }).encode('utf-8')
+        mock_response._content = json.dumps(
+            {"featureFlags": {"flag1": True}, "featureFlagPayloads": {}, "errorsWhileComputingFlags": False}
+        ).encode("utf-8")
 
-        with mock.patch('posthog.request._session.post', return_value=mock_response):
+        with mock.patch("posthog.request._session.post", return_value=mock_response):
             response = decide("fake_key", "fake_host")
             self.assertEqual(response["featureFlags"], {"flag1": True})