test_array_utils.py 文件源码

def test_generic(self):
        """ Test that NaNs are replaced with a fill value """
        with np.errstate(divide='ignore', invalid='ignore'):
            vals = nan_to_num(np.array([0])/0., fill_value = 14)
        self.assertEqual(vals[0], 14)