Fix tests to use BatchSupabaseClient

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: MaxGhenis

scripts/load_pe_targets_batch.py

@@ -137,6 +137,18 @@ def _upsert_single(self, table: str, record: Dict, on_conflict: str):
         headers = {**self.headers, "Prefer": "return=representation"}
         self._request("POST", url, headers=headers, json=record)
 
+    def batch_upsert_strata(self, strata: List[Dict], return_mapping: bool = False) -> Any:
+        """Batch upsert strata with optional name->id mapping."""
+        result = self.batch_upsert("strata", strata, "name,jurisdiction")
+        if return_mapping:
+            return {(r["name"], r["jurisdiction"]): r["id"] for r in result}
+        return result
+
+    def batch_upsert_targets(self, targets: List[Dict], chunk_size: int = 500) -> List[Dict]:
+        """Batch upsert targets."""
+        return self.batch_upsert("targets", targets, "source_id,stratum_id,variable,period",
+                                 chunk_size=chunk_size)
+
 
 @dataclass
 class TargetCollector:

tests/test_supabase_batch.py

@@ -14,9 +14,9 @@
 import responses
 import json
 
-# Import will fail until we implement the class
+# Import the batch client
 try:
-    from scripts.load_pe_targets_to_supabase import SupabaseClient
+    from scripts.load_pe_targets_batch import BatchSupabaseClient as SupabaseClient
 except ImportError:
     SupabaseClient = None