ARROW-27 Test non-ASCII UTF-8 keys schema field names (#48)

Author: blink1073

bindings/python/test/test_bson.py

@@ -103,11 +103,32 @@ def test_simple(self):
 class TestUnsupportedDataType(TestBsonToArrowConversionBase):
 
     def test_simple(self):
-
         schema = Schema({'_id': ObjectId,
                          'data': int64(),
                          'fake':  pyarrow.float16() })
         msg =  ("Unsupported data type in schema for field " +
                 '"fake" of type "halffloat"')
         with self.assertRaisesRegex(ValueError, msg):
             PyMongoArrowContext.from_schema(schema)
+
+
+class TestNonAsciiFieldName(TestBsonToArrowConversionBase):
+
+    def setUp(self):
+        self.schema = Schema({'_id': ObjectId,
+                              'dätá': int64()})
+        self.context = PyMongoArrowContext.from_schema(
+            self.schema)
+
+    def test_simple(self):
+        ids = [ObjectId() for i in range(4)]
+        docs = [{'_id': ids[0], 'dätá': 10 },
+                {'_id': ids[1], 'dätá': 20 },
+                {'_id': ids[2], 'dätá': 30 },
+                {'_id': ids[3], 'dätá': 40 }]
+        as_dict = {
+            '_id': [oid.binary for oid in ids] ,
+            'dätá': [10, 20, 30, 40]
+        }
+
+        self._run_test(docs, as_dict)