Fixed numpy array types, assertion now works correctly.