Added the test for DecisionTreeLearner for the restaurant example given in the book

agarwl · agarwl · commit 7ec1d88d1f8d · 2017-03-01T23:39:58.000+05:30
diff --git a/tests/test_learning.py b/tests/test_learning.py
@@ -3,7 +3,7 @@
 from learning import (
     parse_csv, weighted_mode, weighted_replicate,
     rms_error, manhattan_distance, mean_boolean_error, mean_error,
-    information_content
+    information_content, DecisionTreeLearner, restaurant
 )
 from utils import rounder
 
@@ -55,5 +55,20 @@ def test_information_content():
     # Entropy of a biased coin with heads probability 0.99
     assert rounder(information_content([0.99,0.01])) == 0.0808
 
+def test_DecisionTreeLearner():
+    assert(DecisionTreeLearner(restaurant).display ==
+        '''Test Patrons
+         Patrons = Some ==> RESULT = Yes
+         Patrons = None ==> RESULT = No
+         Patrons = Full ==> Test Type
+            Type = Thai ==> Test Fri/Sat
+                Fri/Sat = No ==> RESULT = No
+                Fri/Sat = Yes ==> RESULT = Yes
+            Type = French ==> RESULT = No
+            Type = Italian ==> RESULT = No
+            Type = Burger ==> Test Hungry
+                Hungry = No ==> RESULT = No
+                Hungry = Yes ==> RESULT = Yes''')
+
 if __name__ == '__main__':
     pytest.main()
