You are comparing floats to int in the `assert_equal` below. It would slightly safer to simply use `arange(12)` in place of `arange(12.)`.