Add new tests for linear algebra