import tagnews

class TestCrimetype():
    @classmethod
    def setup_method(cls):
        cls.model = tagnews.CrimeTags()


    def test_tagtext(self):
        self.model.tagtext('This is example article text')


    def test_tagtext_proba(self):
        article = 'Murder afoul, someone has been shot!'
        probs = self.model.tagtext_proba(article)
        max_prob = probs.max()
        max_type = probs.idxmax()
        tags = self.model.tagtext(article,
                                  prob_thresh=max_prob-0.001)
        assert max_type in tags

Add a used test.