uint handling

Author: flexatone

src/_arraykit.c

@@ -6131,6 +6131,12 @@ AK_TM_transfer(TriMapObject* tm,
                 case NPY_INT8:
                     TRANSFER_SCALARS(npy_int32, npy_int8); // to, from
                     break;
+                case NPY_UINT16:
+                    TRANSFER_SCALARS(npy_int32, npy_uint16); // to, from
+                    break;
+                case NPY_UINT8:
+                    TRANSFER_SCALARS(npy_int32, npy_uint8); // to, from
+                    break;
             }
             break;
         case NPY_INT16:

test/test_tri_map.py

@@ -579,3 +579,27 @@ def test_tri_map_map_uint_d(self) -> None:
         del dst
         self.assertEqual(post_dst.dtype, np.dtype(np.int16))
         self.assertEqual(post_dst.tolist(), [17, 20, 20, 8, 8, 7])
+
+
+    def test_tri_map_map_uint_e(self) -> None:
+        src = np.array([0, 20, 8, 8], dtype=np.uint8)
+        dst = np.array([7, 20, 20, 8], dtype=np.uint8)
+
+        # full outer
+        tm = TriMap(len(src), len(dst))
+        tm.register_one(0, -1)
+        tm.register_many(1, np.array([1, 2], dtype=np.dtype(np.int64)))
+        tm.register_one(2, 3)
+        tm.register_one(3, 3)
+        tm.register_unmatched_dst()
+
+        # import ipdb; ipdb.set_trace()
+        post_src = tm.map_src_fill(src, 17, np.dtype(np.int32))
+        del src
+        self.assertEqual(post_src.tolist(), [0, 20, 20, 8, 8, 17])
+        self.assertEqual(post_src.dtype, np.dtype(np.int32))
+
+        post_dst = tm.map_dst_fill(dst, 17, np.dtype(np.int32))
+        del dst
+        self.assertEqual(post_dst.dtype, np.dtype(np.int32))
+        self.assertEqual(post_dst.tolist(), [17, 20, 20, 8, 8, 7])