Add support for column specific fillna to viz (#3066)

Fix #3029
This commit is contained in:
Riccardo Magliocchetti
2017-09-28 02:51:04 +02:00
committed by Maxime Beauchemin
parent f829b486d1
commit f438ccbcb1
3 changed files with 33 additions and 1 deletions

View File

@@ -767,5 +767,16 @@ class CoreTests(SupersetTestCase):
assert 'language' in resp
self.logout();
def test_viz_get_fillna_for_columns(self):
slc = self.get_slice("Girls", db.session)
q = slc.viz.query_obj()
results = slc.viz.datasource.query(q)
fillna_columns = slc.viz.get_fillna_for_columns(results.df.columns)
self.assertDictEqual(
fillna_columns,
{'name': ' NULL', 'sum__num': 0}
)
if __name__ == '__main__':
unittest.main()