Add a minor extra tests

tskit-dev · Jul 24, 2023 · 83ffc1e · 83ffc1e
1 parent 7400bc6
commit 83ffc1e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_functions.py b/tests/test_functions.py
@@ -1838,7 +1838,7 @@ def test_node_times(self):
 
     def test_fails_unconstrained(self):
         ts = utility_functions.two_tree_mutation_ts()
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match="must be tsdated"):
             nodes_time_unconstrained(ts)
 
 
@@ -1852,9 +1852,14 @@ def test_no_sites(self):
         with pytest.raises(ValueError):
             tsdate.sites_time_from_ts(ts)
 
+    def test_undated(self):
+        ts = utility_functions.two_tree_mutation_ts()
+        with pytest.raises(ValueError, match="Try calling"):
+            tsdate.sites_time_from_ts(ts, unconstrained=True)
+
     def test_node_selection_param(self):
         ts = utility_functions.two_tree_mutation_ts()
-        with pytest.raises(ValueError):
+        with pytest.raises(ValueError, match="node_selection parameter"):
             tsdate.sites_time_from_ts(ts, node_selection="sibling")
 
     def test_sites_time_insideoutside(self):