Added multiplicative distributivity over addition test

1 files changed, 23 insertions, 0 deletions

diff --git a/Week7/test/polynomial/PolynomialTest.java b/Week7/test/polynomial/PolynomialTest.java
index 8560dfb..c00f165 100644
--- a/Week7/test/polynomial/PolynomialTest.java
+++ b/Week7/test/polynomial/PolynomialTest.java
@@ -123,6 +123,29 @@ public class PolynomialTest {
         Polynomial b_a = new Polynomial(b);
         b_a.times(a);
         assertEquals(true, a_b.equals(b_a));
+        
+        // Test left distributivity
+        Polynomial b_plus_c = new Polynomial(b);
+        b_plus_c.plus(c);
+        Polynomial a_times_b_plus_c = new Polynomial(a);
+        a_times_b_plus_c.times(b_plus_c);
+        Polynomial a_times_c = new Polynomial(a);
+        a_times_c.times(c);
+        Polynomial a_times_b_plus_a_times_c = new Polynomial(a);
+        a_times_b_plus_a_times_c.times(b);
+        a_times_b_plus_a_times_c.plus(a_times_c);
+        assertEquals(true, a_times_b_plus_c.equals(a_times_b_plus_a_times_c));
+        
+        // Test right distributivity
+        Polynomial b_plus_c_times_a = new Polynomial(b);
+        b_plus_c_times_a.plus(c);
+        b_plus_c_times_a.times(a);
+        Polynomial c_times_a = new Polynomial(c);
+        c_times_a.times(a);
+        Polynomial b_times_a_plus_c_times_a = new Polynomial(b);
+        b_times_a_plus_c_times_a.times(a);
+        b_times_a_plus_c_times_a.plus(c_times_a);
+        assertEquals(true, b_plus_c_times_a.equals(b_times_a_plus_c_times_a));
     }
 
     /**