I'd use ```np.allclose``` rather than ```np.testing.assert_almost_equal```.