I think you want to add an import line such as:

```python
from numpy.testing import assert_array_equal, assert_array_almost_equal
```