Skip to content

Commit

Permalink
fix raise regex
Browse files Browse the repository at this point in the history
  • Loading branch information
taylorfturner committed Sep 19, 2022
1 parent c1f5ca1 commit ce1dc35
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion dataprofiler/tests/labelers/test_column_name_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,13 @@ def test_predict(self, mock_stdout):

# `show_confidences` is disabled currently
# should raise error if set to `True`
with self.assertRaises(NotImplementedError):
with self.assertRaisesRegex(
Warning,
"""`show_confidences` parameter is disabled
for MVP implementation. Due to the requirement
of having the data point in the post processor.
Note: Confidence values are returned by default.""",
):
model.predict(data=self.data, show_confidences=True)

# clear stdout
Expand Down

0 comments on commit ce1dc35

Please sign in to comment.