Commit bce808ca authored by Sascha Herzinger's avatar Sascha Herzinger
Browse files

Adding a couple of additional info to histogram analysis output

parent fa7b4cbc
......@@ -65,5 +65,8 @@ class HistogramTask(AnalyticTask):
'variance': variance
}
return {
'stats': stats
'stats': stats,
'subsets': subsets,
'categories': categories,
'label': df['feature'].tolist()[0]
}
......@@ -35,7 +35,8 @@ class TestHistogramTask:
subsets=[],
data=df,
categories=[cat_df])
assert 'stats' in result
assert all([key in result for key in
['stats', 'subsets', 'categories', 'label']])
assert 'A' in result['stats']
assert 'B' in result['stats']
assert 0 in result['stats']['A']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment