Addition of match operator about tablite HOT 5 CLOSED

import numpy as np
from tablite import Table
from tablite.reindex import reindex 
from tablite.utils import unique_name

def match(T, other, *criteria):
    """
    matches value from T with other
    """
    # analyze
    for a,_,b in criteria:
        assert _ == "=="
        
        indices = other[b].index()  # {'a':[0,2], 'b': [1,4], 'c': [3]}
        index = np.full((len(T), ), -1, dtype=int)
        for ix,v in enumerate(T[a][:]):
            if index[ix] == -1:
                index[ix] = indices[v][0]
    # write
    result = T.copy()
    second = reindex(other, index)
    for name in other.columns:
        revised_name = unique_name(name, result.columns)
        result[revised_name] = second[name]
    
    return result

normalized_bom = Table({'bom_id': [1, 2, 3, 4, 5, 6, 7, 8, 9], 'partial_of': [1, 2, 3, 4, 5, 6, 7, 4, 6], 'sku': ['A', 'irrelevant', 'empty carton', 'pkd carton', 'empty pallet', 'pkd pallet', 'pkd irrelevant', 'ppkd carton', 'ppkd pallet'], 'material_id': [None, None, None, 3, None, 5, 3, 3, 5], 'quantity': [1, 1, 1, 1, 1, 1, 1, 1, 1], 'L': [0.25, 0.1, 0.6, 0.6, 1.2, 1.2, 0.6, 0.6, 1.2], 'W': [0.15, 0.1, 0.4, 0.4, 1.0, 1.0, 0.4, 0.4, 1.0], 'H': [0.15, 0.1, 0.3, 0.3, 0.15, 0.15, 0.3, 0.3, 0.15], 'M': [0.13, 1.0, 0.3, 0.3, 12.5, 12.5, 0.3, 0.3, 12.5], 'L2': [None, None, 0.6, 0.6, 1.2, 1.2, 0.6, 0.6, 1.2], 'W2': [None, None, 0.4, 0.4, 1.0, 1.0, 0.4, 0.4, 1.0], 'H2': [None, None, 0.3, 0.3, 1.8, 1.8, 0.3, 0.3, 1.8], 'M2': [None, None, 40.0, 40.0, 1200.0, 1200.0, 40.0, 40.0, 1200.0], 'packtype': ['item', 'item', 'material', 'carton', 'material', 'pallet', 'carton', 'carton', 'pallet'], 'properties': [None, None, None, None, None, None, None, None, None]})
accepts = Table({'bom_id': [6], 'sku': ['pkd pallet'], 'supply_time': [np.timedelta64(3600000000,'us')], 'moq': [1]})
products_lookup = normalized_bom.lookup(accepts, ("bom_id", "==", "bom_id"), ("partial_of", "==", "bom_id"), all=False)
products = products_lookup.all(bom_id_1=lambda x: x is not None)

products_matched = match(normalized_bom, accepts, ("bom_id", "==", "bom_id"), ("partial_of", "==", "bom_id"))

assert products.to_dict() == products_matched.to_dict()