add boolean array to unit tests

Author: rasbt

mlxtend/frequent_patterns/fpgrowth.py

@@ -5,8 +5,6 @@
 
 import math
 import itertools
-import numpy as np
-import pandas as pd
 from ..frequent_patterns import fpcommon as fpc
 
 

mlxtend/frequent_patterns/tests/test_apriori.py

@@ -5,10 +5,27 @@
 # License: BSD 3 clause
 
 import unittest
+import numpy as np
 from mlxtend.frequent_patterns.tests.test_fpbase import FPTestAll
 from mlxtend.frequent_patterns import apriori
 
 
 class TestApriori(unittest.TestCase, FPTestAll):
     def setUp(self):
         FPTestAll.setUp(self, apriori)
+
+
+class TestApriori2(unittest.TestCase, FPTestAll):
+    def setUp(self):
+        one_ary = np.array(
+            [[False, False, False, True, False, True, True, True, True,
+              False, True],
+             [False, False, True, True, False, True, False, True, True,
+              False, True],
+             [True, False, False, True, False, True, True, False, False,
+              False, False],
+             [False, True, False, False, False, True, True, False, False,
+              True, True],
+             [False, True, False, True, True, True, False, False, True,
+              False, False]])
+        FPTestAll.setUp(self, apriori, one_ary=one_ary)

mlxtend/frequent_patterns/tests/test_fpbase.py

@@ -16,14 +16,18 @@ class FPTestBase(object):
     setup and tests common to all methods (e.g., error for improper input)
     """
 
-    def setUp(self, fpalgo):
-        self.one_ary = np.array(
-            [[0, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1],
+    def setUp(self, fpalgo, one_ary=None):
+        if one_ary is None:
+            self.one_ary = np.array(
+               [[0, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1],
                 [0, 0, 1, 1, 0, 1, 0, 1, 1, 0, 1],
                 [1, 0, 0, 1, 0, 1, 1, 0, 0, 0, 0],
                 [0, 1, 0, 0, 0, 1, 1, 0, 0, 1, 1],
                 [0, 1, 0, 1, 1, 1, 0, 0, 1, 0, 0]])
 
+        else:
+            self.one_ary = one_ary
+
         self.cols = ['Apple', 'Corn', 'Dill', 'Eggs', 'Ice cream',
                      'Kidney Beans', 'Milk',
                      'Nutmeg', 'Onion', 'Unicorn', 'Yogurt']
@@ -100,8 +104,8 @@ def test_with_fill_values(fill_value):
 
 
 class FPTestAll(FPTestBase):
-    def setUp(self, fpalgo):
-        FPTestBase.setUp(self, fpalgo)
+    def setUp(self, fpalgo, one_ary=None):
+        FPTestBase.setUp(self, fpalgo, one_ary=one_ary)
 
     def test_default(self):
         res_df = self.fpalgo(self.df)
@@ -131,8 +135,8 @@ def test_max_len(self):
 
 
 class FPTestMaximal(FPTestBase):
-    def setUp(self, fpalgo):
-        FPTestBase.setUp(self, fpalgo)
+    def setUp(self, fpalgo, one_ary=None):
+        FPTestBase.setUp(self, fpalgo, one_ary=one_ary)
 
     def test_default(self):
         res_df = self.fpalgo(self.df)

mlxtend/frequent_patterns/tests/test_fpgrowth.py

@@ -1,8 +1,25 @@
 import unittest
+import numpy as np
 from mlxtend.frequent_patterns.tests.test_fpbase import FPTestAll
 from mlxtend.frequent_patterns import fpgrowth
 
 
 class TestFPGrowth(unittest.TestCase, FPTestAll):
     def setUp(self):
         FPTestAll.setUp(self, fpgrowth)
+
+
+class TestFPGrowth2(unittest.TestCase, FPTestAll):
+    def setUp(self):
+        one_ary = np.array(
+            [[False, False, False, True, False, True, True, True, True,
+              False, True],
+             [False, False, True, True, False, True, False, True, True,
+              False, True],
+             [True, False, False, True, False, True, True, False, False,
+              False, False],
+             [False, True, False, False, False, True, True, False, False,
+              True, True],
+             [False, True, False, True, True, True, False, False, True,
+              False, False]])
+        FPTestAll.setUp(self, fpgrowth, one_ary=one_ary)

mlxtend/frequent_patterns/tests/test_fpmax.py

@@ -1,8 +1,25 @@
 import unittest
+import numpy as np
 from mlxtend.frequent_patterns.tests.test_fpbase import FPTestMaximal
 from mlxtend.frequent_patterns import fpmax
 
 
 class TestFPMax(unittest.TestCase, FPTestMaximal):
     def setUp(self):
         FPTestMaximal.setUp(self, fpmax)
+
+
+class TestFPMax2(unittest.TestCase, FPTestMaximal):
+    def setUp(self):
+        one_ary = np.array(
+            [[False, False, False, True, False, True, True, True, True,
+              False, True],
+             [False, False, True, True, False, True, False, True, True,
+              False, True],
+             [True, False, False, True, False, True, True, False, False,
+              False, False],
+             [False, True, False, False, False, True, True, False, False,
+              True, True],
+             [False, True, False, True, True, True, False, False, True,
+              False, False]])
+        FPTestMaximal.setUp(self, fpmax, one_ary=one_ary)