add float part

johnlinvc · Mar 23, 2019 · 44cb2c1 · 44cb2c1
1 parent be6d148
commit 44cb2c1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/autodiff/numeric.rb b/lib/autodiff/numeric.rb
@@ -17,3 +17,9 @@ class Integer
   dual_op(:+)
   dual_op(:-)
 end
+
+class Float
+  extend DualNumConvertible
+  dual_op(:+)
+  dual_op(:-)
+end
diff --git a/test/autodiff_test.rb b/test/autodiff_test.rb
@@ -21,6 +21,10 @@ def test_gradient_can_do_1_minus_x
     assert_equal -1, Autodiff.gradient(0) { |x| 1 - x }
   end
 
+  def test_gradient_can_do_1_point_5_minus_x
+    assert_equal -1, Autodiff.gradient(0) { |x| 1.5 - x }
+  end
+
   def test_gradient_can_handle_x_square
     assert_equal 2, Autodiff.gradient(1) { |x| x*x}
   end