Clever! It's clearer now, I didn't think of using `assert_array_almost_equal` this way.