Add a test for subscript assignment