fix python tests

funbringer · funbringer · commit beaf8dda3966 · 2017-02-10T07:02:39.000+03:00
diff --git a/tests/python/partitioning_test.py b/tests/python/partitioning_test.py
@@ -773,8 +773,8 @@ def test_pg_dump(self):
 				end
 				$$ language plpgsql;
 			""")
-			con.execute('select set_init_callback(\'range_partitioned\', \'init_partition_stub_callback\')')
-			con.execute('select set_init_callback(\'hash_partitioned\', \'init_partition_stub_callback\')')
+			con.execute('select set_init_callback(\'range_partitioned\', \'init_partition_stub_callback(jsonb)\')')
+			con.execute('select set_init_callback(\'hash_partitioned\', \'init_partition_stub_callback(jsonb)\')')
 
 			# turn off enable_parent option
 			con.execute('select set_enable_parent(\'range_partitioned\', false)')
