def test_default(self):
res_df = fpmax(self.df)
expect = pd.DataFrame(
[[0.6, frozenset([5, 6])], [0.6, frozenset([5, 10])],
[0.6, frozenset([3, 5, 8])]],
columns=['support', 'itemsets'])
compare_dataframes(res_df, expect)
def test_output(self):
res_df = fpmax(self.df, min_support=0.001, use_colnames=True)
expect = pd.DataFrame(
[[0.25, frozenset(['a'])], [0.25, frozenset(['b'])],
[0.25, frozenset(['c', 'd'])], [0.25, frozenset(['e'])]],
columns=['support', 'itemsets'])
compare_dataframes(res_df, expect)
def test_output(self):
res_df = self.fpalgo(self.df, min_support=0.01, use_colnames=True)
expect = pd.DataFrame(
[[0.3333333333333333, frozenset(['a', 'b'])],
[0.3333333333333333, frozenset(['a', 'c'])]],
columns=['support', 'itemsets'])
compare_dataframes(res_df, expect)