Added test cases

casact · Mar 26, 2024 · d06ac9b · d06ac9b
1 parent fddc0a6
commit d06ac9b
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/chainladder/core/tests/test_display.py b/chainladder/core/tests/test_display.py
@@ -3,18 +3,18 @@
 
 
 def test_heatmap_render(raa):
-    """ The heatmap method should render correctly given the sample."""
-    return raa.heatmap()
+    """The heatmap method should render correctly given the sample."""
+    assert raa.heatmap()
 
 
-def test_to_frame(raa):
-    try:
-        cl.Chainladder().fit(raa).cdf_.to_frame()
-        cl.Chainladder().fit(raa).cdf_.to_frame(origin_as_datetime=False)
-        cl.Chainladder().fit(raa).cdf_.to_frame(origin_as_datetime=True)
-        cl.Chainladder().fit(raa).ultimate_.to_frame()
-        cl.Chainladder().fit(raa).ultimate_.to_frame(origin_as_datetime=False)
-        cl.Chainladder().fit(raa).ultimate_.to_frame(origin_as_datetime=True)
+def test_empty_triangle():
+    assert cl.Triangle()
+
 
-    except:
-        assert False
+def test_to_frame(raa):
+    assert cl.Chainladder().fit(raa).cdf_.to_frame()
+    assert cl.Chainladder().fit(raa).cdf_.to_frame(origin_as_datetime=False)
+    assert cl.Chainladder().fit(raa).cdf_.to_frame(origin_as_datetime=True)
+    assert cl.Chainladder().fit(raa).ultimate_.to_frame()
+    assert cl.Chainladder().fit(raa).ultimate_.to_frame(origin_as_datetime=False)
+    assert cl.Chainladder().fit(raa).ultimate_.to_frame(origin_as_datetime=True)