Add unittest

ecoscope/io/earthranger.py

@@ -1040,10 +1040,11 @@ def post_source(
         response = self._post("sources", payload=payload)
         return pd.DataFrame([response])
 
-    def post_source_provider(
+    def post_sourceproviders(
         self,
         provider_key: str,
         display_name: str,
+        additional: typing.Dict = {},
         **kwargs,
     ) -> pd.DataFrame:
         """
@@ -1060,6 +1061,7 @@ def post_source_provider(
         payload = {
             "provider_key": provider_key,
             "display_name": display_name,
+            "additional": additional,
         }
 
         if kwargs:

tests/test_earthranger_io.py

@@ -6,11 +6,10 @@
 import pandas as pd
 import pytest
 import pytz
+from erclient import ERClientException
 from shapely.geometry import Point
 
 import ecoscope
-from erclient import ERClientException
-
 from ecoscope.io.earthranger import EarthRangerIO
 from ecoscope.io.earthranger_utils import TIME_COLS
 
@@ -205,6 +204,13 @@ def test_get_patrol_events_empty(patrols_mock, er_io):
     assert events.empty
 
 
+def test_post_sourceproviders(er_io):
+    response = er_io.post_sourceproviders(provider_key="test_provider_key", display_name="Test Provider Key")
+    assert "id" in response
+    assert "provider_key" in response
+    assert "display_name" in response
+
+
 def test_post_observations(er_io):
     observations = gpd.GeoDataFrame.from_dict(
         [