unit test

src/main/java/com/github/cschen1205/fuzzylogic/Rule.java

@@ -18,11 +18,11 @@ public Clause getConsequent() {
             return mConsequent;
         }
 
-        public void setmConsequent(Clause value){
+        public void setConsequent(Clause value){
             mConsequent = value;
         }
 
-        public void AddAntecedent(Clause antecedent)
+        public void addAntecedent(Clause antecedent)
         {
             mAntecedents.add(antecedent);
         }

src/main/java/com/github/cschen1205/fuzzylogic/RuleInferenceEngine.java

@@ -11,7 +11,7 @@ public class RuleInferenceEngine
         Map<String, FuzzySet> mWorkingMemory = new HashMap<>();
         List<Rule> mRules = new ArrayList<>();
 
-        public void AddFuzzySet(String variable, FuzzySet set)
+        public void addFuzzySet(String variable, FuzzySet set)
         {
             mVariables.add(variable);
             mWorkingMemory.put(variable, set);
@@ -23,17 +23,17 @@ public int getFuzzySetCount()
 
         }
 
-        public void AddRule(Rule fr)
+        public void addRule(Rule fr)
         {
             mRules.add(fr);
         }
 
-        public void SetVariable(String variable, double crispValue)
+        public void setVariable(String variable, double crispValue)
         {
             mWorkingMemory.get(variable).setX(crispValue);
         }
 
-        public double GetVariable(String variable)
+        public double getVariable(String variable)
         {
             return mWorkingMemory.get(variable).getX();
         }

src/test/java/com/github/cschen1205/fuzzylogic/RuleInferenceEngineUnitTest.java

@@ -0,0 +1,124 @@
+package com.github.cschen1205.fuzzylogic;
+
+import com.github.cschen1205.fuzzylogic.memberships.FuzzyGrade;
+import com.github.cschen1205.fuzzylogic.memberships.FuzzyReverseGrade;
+import com.github.cschen1205.fuzzylogic.memberships.FuzzyTriangle;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.testng.annotations.Test;
+
+import static org.testng.Assert.*;
+
+/**
+ * Created by memeanalytics on 3/6/17.
+ */
+public class RuleInferenceEngineUnitTest {
+
+    private static final Logger logger = LoggerFactory.getLogger(RuleInferenceEngineUnitTest.class);
+
+    @Test
+    public void test_simple(){
+        RuleInferenceEngine rie=new RuleInferenceEngine();
+
+        FuzzySet output=new FuzzySet("Output", -100, 100, 1);
+        output.addMembership("Cooling", new FuzzyReverseGrade(-50, 0));
+        output.addMembership("Zero", new FuzzyTriangle(-50, 0, 50));
+        output.addMembership("Heating", new FuzzyGrade(0, 50));
+        rie.addFuzzySet(output.getName(), output);
+
+        FuzzySet tempError=new FuzzySet("Temperature Error", -4, 4, 0.05);
+        tempError.addMembership("Negative", new FuzzyReverseGrade(-2, 0));
+        tempError.addMembership("Zero", new FuzzyTriangle(-2, 0, 2));
+        tempError.addMembership("Positive", new FuzzyGrade(0, 2));
+        rie.addFuzzySet(tempError.getName(), tempError);
+
+
+        FuzzySet tempErrorDot=new FuzzySet("Temperature Error", -10, 10, 0.1);
+        tempErrorDot.addMembership("Negative", new FuzzyReverseGrade(-5, 0));
+        tempErrorDot.addMembership("Zero", new FuzzyTriangle(-5, 0, 5));
+        tempErrorDot.addMembership("Positive", new FuzzyGrade(0, 5));
+        rie.addFuzzySet(tempErrorDot.getName(), tempErrorDot);
+
+		    /*
+		    1. If (e < 0) AND (er < 0) then Cool 0.5 & 0.0 = 0.0
+		    2. If (e = 0) AND (er < 0) then Heat 0.5 & 0.0 = 0.0
+		    3. If (e > 0) AND (er < 0) then Heat 0.0 & 0.0 = 0.0
+		    4. If (e < 0) AND (er = 0) then Cool 0.5 & 0.5 = 0.5
+		    5. If (e = 0) AND (er = 0) then No_Chng 0.5 & 0.5 = 0.5
+		    6. If (e > 0) AND (er = 0) then Heat 0.0 & 0.5 = 0.0
+		    7. If (e < 0) AND (er > 0) then Cool 0.5 & 0.5 = 0.5
+		    8. If (e = 0) AND (er > 0) then Cool 0.5 & 0.5 = 0.5
+		    9. If (e > 0) AND (er > 0) then Heat 0.0 & 0.5 = 0.0
+		    */
+
+        Rule rule=new Rule("Rule 1");
+        rule.addAntecedent(new Clause(tempError, "Is", "Negative"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Negative"));
+        rule.setConsequent(new Clause(output, "Is", "Cooling"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 2");
+        rule.addAntecedent(new Clause(tempError, "Is", "Zero"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Negative"));
+        rule.setConsequent(new Clause(output, "Is", "Heating"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 3");
+        rule.addAntecedent(new Clause(tempError, "Is", "Positive"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Negative"));
+        rule.setConsequent(new Clause(output, "Is", "Heating"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 4");
+        rule.addAntecedent(new Clause(tempError, "Is", "Negative"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Zero"));
+        rule.setConsequent(new Clause(output, "Is", "Cooling"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 5");
+        rule.addAntecedent(new Clause(tempError, "Is", "Zero"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Zero"));
+        rule.setConsequent(new Clause(output, "Is", "Zero"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 6");
+        rule.addAntecedent(new Clause(tempError, "Is", "Positive"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Zero"));
+        rule.setConsequent(new Clause(output, "Is", "Heating"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 7");
+        rule.addAntecedent(new Clause(tempError, "Is", "Negative"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Positive"));
+        rule.setConsequent(new Clause(output, "Is", "Cooling"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 8");
+        rule.addAntecedent(new Clause(tempError, "Is", "Zero"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Positive"));
+        rule.setConsequent(new Clause(output, "Is", "Cooling"));
+        rie.addRule(rule);
+
+        rule=new Rule("Rule 9");
+        rule.addAntecedent(new Clause(tempError, "Is", "Positive"));
+        rule.addAntecedent(new Clause(tempErrorDot, "Is", "Positive"));
+        rule.setConsequent(new Clause(output, "Is", "Heating"));
+        rie.addRule(rule);
+
+        tempError.setX(-1.0);
+        logger.info("For Temperature Error: {}", tempError.getX());
+        logger.info("Negative: "+tempError.GetMembership("Negative").degree(-1.0));
+        logger.info("Zero: "+tempError.GetMembership("Zero").degree(-1.0));
+        logger.info("Positive: "+tempError.GetMembership("Positive").degree(-1.0));
+
+        tempErrorDot.setX(2.5);
+        logger.info("For Temperature Error Dot: {}", tempErrorDot.getX());
+        logger.info("Negative: "+tempErrorDot.GetMembership("Negative").degree(2.5));
+        logger.info("Zero: "+tempErrorDot.GetMembership("Zero").degree(2.5));
+        logger.info("Positive: "+tempErrorDot.GetMembership("Positive").degree(2.5));
+
+        rie.Infer(output);
+
+        logger.info("output: {} = {}", output.getName(), output.getX());
+    }
+}

src/test/resources/log4j.properties

@@ -0,0 +1,10 @@
+# Set root logger level to DEBUG and its only appender to A1.
+log4j.rootLogger=DEBUG, A1
+
+# A1 is set to be a ConsoleAppender.
+log4j.appender.A1=org.apache.log4j.ConsoleAppender
+
+# A1 uses PatternLayout.
+log4j.appender.A1.layout=org.apache.log4j.PatternLayout
+log4j.appender.A1.layout.ConversionPattern=%-5p %c %x - %m%n
+