Add test for flat_map with custom kernel