Value equality for MortalityTable types (#126)

JuliaActuary · Mar 3, 2023 · 616a20f · 616a20f
1 parent aa662ae
commit 616a20f
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -23,4 +23,6 @@ docs/site/
 # environment.
 Manifest.toml
 
+.vscode
+
 .DS_Store
diff --git a/src/MortalityTable.jl b/src/MortalityTable.jl
@@ -108,6 +108,16 @@ end
 Base.getindex(u::UltimateTable, x) = u.ultimate[x]
 Base.lastindex(u::UltimateTable) = lastindex(u.ultimate)
 
+Base.:(==)(tbl1::UltimateTable, tbl2::UltimateTable) = tbl1.metadata == tbl2.metadata && isequal(tbl1.ultimate, tbl2.ultimate)
+function Base.:(==)(tbl1::SelectUltimateTable, tbl2::SelectUltimateTable)
+    return (
+        tbl1.metadata == tbl2.metadata && 
+        isequal(tbl1.ultimate, tbl2.ultimate) &&
+        isequal(tbl1.select, tbl2.select)
+    )
+end
+
+
 
 function MortalityTable(select, ultimate; metadata = TableMetaData())
     return SelectUltimateTable(select, ultimate, metadata)

diff --git a/test/MortalityTables.jl b/test/MortalityTables.jl
@@ -141,9 +141,17 @@
 
         end
 
-
-
-
+        @testset "value equality" begin
+            ult_table_name = "1980 CSO Basic Table – Male, ANB"
+            cso1980_a = MortalityTables.table(ult_table_name)
+            cso1980_b = MortalityTables.table(ult_table_name)
+            @test cso1980_a == cso1980_b
+
+            select_table_name = "2001 CSO Super Preferred Select and Ultimate - Male Nonsmoker, ANB"
+            cso2001_a = MortalityTables.table(select_table_name)
+            cso2001_b = MortalityTables.table(select_table_name)
+            @test cso2001_a == cso2001_b
+        end
 
     end
 end