Initial commit

DecisionTreeTest.java

@@ -0,0 +1,74 @@
+
+import static org.junit.Assert.*;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+public class DecisionTreeTest {
+
+    private static final String[][] known_data = {
+        {"sonnig", "heiß", "hoch", "schwach", "ja"},
+        {"sonnig", "heiß", "hoch", "stark", "nein"},
+        {"bewölkt", "heiß", "hoch", "schwach", "ja"},
+        {"regnerisch", "mild", "hoch", "schwach", "nein"},
+        {"regnerisch", "kalt", "normal", "schwach", "ja"},
+        {"regnerisch", "kalt", "normal", "stark", "ja"},
+        {"bewölkt", "mild", "hoch", "stark", "ja"},
+        {"sonnig", "mild", "hoch", "schwach", "ja"},
+        {"sonnig", "kalt", "normal", "schwach", "ja"},
+        {"regnerisch", "mild", "normal", "schwach", "ja"},
+        {"sonnig", "mild", "normal", "stark", "nein"},
+        {"bewölkt", "heiß", "normal", "schwach", "ja"},
+        {"bewölkt", "kalt", "normal", "stark", "ja"},
+        {"regnerisch", "mild", "hoch", "stark", "nein"}
+    };
+
+    private static final String[][] unknown_data = {
+        {"regnerisch", "heiß", "hoch", "schwach", "nein"},
+        {"sonnig", "kalt", "hoch", "stark", "nein"}
+    };
+
+    private DecisionTreeBuilder tree;
+
+    private Dataset[] known_datasets, unknown_datasets;
+
+    @Before
+    public void setUp() {
+        known_datasets = new Dataset[known_data.length];
+        for( int i = 0; i < known_data.length; i++ ) {
+            known_datasets[i] = new Dataset();
+            known_datasets[i].set("vorhersage", known_data[i][0]);
+            known_datasets[i].set("temperatur", known_data[i][1]);
+            known_datasets[i].set("feuchtigkeit", known_data[i][2]);
+            known_datasets[i].set("wind", known_data[i][3]);
+        }
+
+        unknown_datasets = new Dataset[unknown_data.length];
+        for( int i = 0; i < unknown_data.length; i++ ) {
+            unknown_datasets[i] = new Dataset();
+            unknown_datasets[i].set("vorhersage", unknown_data[i][0]);
+            unknown_datasets[i].set("temperatur", unknown_data[i][1]);
+            unknown_datasets[i].set("feuchtigkeit", unknown_data[i][2]);
+            unknown_datasets[i].set("wind", unknown_data[i][3]);
+        }
+
+        tree = new DecisionTreeBuilder();
+    }
+
+    @Test
+    public void testKnownData() {
+        for( int i = 0; i < known_datasets.length; i++ ) {
+            String result = tree.testDataset(known_datasets[i]);
+            assertEquals(known_data[i][4], result);
+        }
+    }
+
+    @Test
+    public void testUnknownData() {
+        for( int i = 0; i < unknown_datasets.length; i++ ) {
+            String result = tree.testDataset(unknown_datasets[i]);
+            assertEquals(unknown_data[i][4], result);
+        }
+    }
+
+}