Move identity tests to a separate file