Add tests for assignment operator