Added new args to load_data and added new test about load_data

ibis/mapd/client.py

@@ -691,7 +691,7 @@ def database(self, name=None):
             )
             return self.database_class(name, new_client)
 
-    def load_data(self, table_name, obj, database=None):
+    def load_data(self, table_name, obj, database=None, *args, **kwargs):
         """
         Wraps the LOAD DATA DDL statement. Loads data into an MapD table by
         physically moving data files.
@@ -704,7 +704,7 @@ def load_data(self, table_name, obj, database=None):
         """
         _database = self.db_name
         self.set_database(database)
-        self.con.load_table(table_name, obj)
+        self.con.load_table(table_name, obj, *args, **kwargs)
         self.set_database(_database)
 
     @property

ibis/mapd/tests/test_client.py

@@ -67,3 +67,22 @@ def test_compile_toplevel():
     result = ibis.mapd.compile(expr)
     expected = 'SELECT sum("foo") AS "sum"\nFROM t0'  # noqa
     assert str(result) == expected
+
+
+def test_load_data(con):
+    import ibis.pandas
+
+    df = pd.DataFrame({
+        'a': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I'],
+        'b': [28., 55., 43, 91, 81, 53, 19, 87, 52]
+    })
+
+    tb = ibis.pandas.connect({'value': df}).table('value')
+
+    try:
+        con.drop_table('value')
+    except Exception:
+        pass
+    con.create_table('value', schema=tb.schema())
+    con.load_data('value', df)
+    con.drop_table('value')