def test_filter_dataframe_4(self):
"""test with top_x=5, threshold=7 (so threshold leaves fewer than 5 results), name=auto"""
test_wc = WorkflowManager([])
test_wc._result = pandas.DataFrame({'data': pandas.Series([i for i in range(10, 0, -1)])})
expected = pandas.DataFrame({'data': pandas.Series([10, 9, 8])})
actual = test_wc._filter_dataframe(top_x=5, threshold=7)
self.assertEqual(len(expected.columns), len(actual.columns))
for i in expected.columns:
self.assertSequenceEqual(list(expected[i].index), list(actual[i].index))
self.assertSequenceEqual(list(expected[i].values), list(actual[i].values))
def test_filter_dataframe_5(self):
"""test with top_x=3, threshold=auto, name='asdf'; many input columns"""
test_wc = WorkflowManager([])
test_wc._result = pandas.DataFrame({('1', 'b'): pandas.Series([i for i in range(10, 0, -1)]),
('1', 'z'): pandas.Series([i for i in range(10, 20)]),
('2', 'e'): pandas.Series([i for i in range(40, 900)])})
expected = pandas.DataFrame({'asdf': pandas.Series([10, 9, 8])})
actual = test_wc._filter_dataframe(top_x=3, name='asdf')
self.assertEqual(len(expected.columns), len(actual.columns))
for i in expected.columns:
self.assertSequenceEqual(list(expected[i].index), list(actual[i].index))
self.assertSequenceEqual(list(expected[i].values), list(actual[i].values))
