Convert linalg test to py.test