def test_init_df_with_na_unit(test_pd_df, tmpdir):
# missing values in the unit column are replaced by an empty string
test_pd_df.loc[1, "unit"] = np.nan
df = IamDataFrame(test_pd_df)
assert df.unit == ["", "EJ/yr"]
# writing to file and importing as pandas returns `nan`, not empty string
file = tmpdir / "na_unit.csv"
df.to_csv(file)
df_csv = pd.read_csv(file)
assert np.isnan(df_csv.loc[1, "Unit"])
IamDataFrame(file) # reading from file as IamDataFrame works
file = tmpdir / "na_unit.xlsx"
df.to_excel(file)
df_excel = pd.read_excel(file, engine="openpyxl")
assert np.isnan(df_excel.loc[1, "Unit"])
IamDataFrame(file) # reading from file as IamDataFrame works
def test_init_df_with_na_unit(test_pd_df, tmpdir):
# missing values in the unit column are replaced by an empty string
test_pd_df.loc[1, 'unit'] = np.nan
df = IamDataFrame(test_pd_df)
assert df.unit == ['', 'EJ/yr']
# writing to file and importing as pandas returns `nan`, not empty string
file = tmpdir / 'na_unit.csv'
df.to_csv(file)
df_csv = pd.read_csv(file)
assert np.isnan(df_csv.loc[1, 'Unit'])
IamDataFrame(file) # reading from file as IamDataFrame works
file = tmpdir / 'na_unit.xlsx'
df.to_excel(file)
df_excel = pd.read_excel(file)
assert np.isnan(df_excel.loc[1, 'Unit'])
IamDataFrame(file) # reading from file as IamDataFrame works
