`assert_array_equal` is better here, because `assert` gets removed when Python is in optimized mode ('Python -O').