Add test for adi-2d