add upsert

Author: davidkhala

frame/davidkhala/data/frame/pandas/__init__.py

@@ -0,0 +1,3 @@
+def upsert(df, primary_key:str, record:dict):
+    df.loc[record[primary_key]] = record
+    return df

frame/tests/pandas_test.py

@@ -15,6 +15,21 @@ def test_duckDB(self):
         conn.register("airports", df_airports)
         conn.close()
 
+class SyntaxTestCase(unittest.TestCase):
+    def test_upsert(self):
+        from davidkhala.data.frame.pandas import upsert
+        key ='id'
+        df = pandas.DataFrame({
+            key: [1, 2],
+            'name': ['Alice', 'Bob'],
+            'score': [85, 90]
+        }).set_index(key)
+        new_record = {'name': 'Charlie', 'score': 95, key:2}
+        upsert(df, key, new_record)
+
+        self.assertEqual(95, df.at[2, 'score'])
+
+
 
 
 if __name__ == '__main__':