Support augmented assign for ndarrays
Add a difficult test case from parakeet