Test delinearization of 2D diagonal matrix