Add test for distributive law transform