add operations

geocrystal · Apr 30, 2024 · 2531804 · 2531804
1 parent 594a437
commit 2531804
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/spec/haversine/distance_spec.cr b/spec/haversine/distance_spec.cr
@@ -16,4 +16,12 @@ describe Haversine::Distance do
       end
     end
   end
+
+  describe "operations" do
+    dist1 = Haversine::Distance.new(2)
+    dist2 = Haversine::Distance.new(1)
+
+    it { (dist1 + dist2).distance.should eq(3) }
+    it { (dist1 - dist2).distance.should eq(1) }
+  end
 end
diff --git a/src/haversine/distance.cr b/src/haversine/distance.cr
@@ -16,5 +16,15 @@ module Haversine
     def <=>(other : Haversine::Distance)
       distance <=> other.distance
     end
+
+    # Adds the value of `self` to *other*.
+    def +(other : Haversine::Distance) : Haversine::Distance
+      Haversine::Distance.new(distance + other.distance)
+    end
+
+    # Removes the value of *other* from `self`.
+    def -(other : Haversine::Distance) : Haversine::Distance
+      Haversine::Distance.new(distance - other.distance)
+    end
   end
 end