def test_get_nan_rep_no_arg(self): config_dict = {'io_output_na_rep': 'nan'} args = MagicMock(io_output_na_rep=None) self.assertTrue(np.isnan(io_lib.get_nan_rep(args, config_dict)))
def test_get_nan_rep_with_nan(self): config_dict = {'io_output_na_rep': '-'} args = MagicMock(io_output_na_rep=['-']) self.assertEqual(io_lib.get_nan_rep(args, config_dict), '-')