class TestStattext(TestCase): def setUp(self): self.stattext = Stattext() @mock.patch.object(Stattext, '_count_words') def test_total_words(self, mock_count_words): self.stattext.total_words = None self.stattext.get_total_words("fake") mock_count_words.assert_called_with("fake") @mock.patch.object(Stattext, '_count_words') def test_count_words_no_called(self, mock_count_words): self.stattext.total_words = 21 self.stattext.get_total_words("fake") self.assertFalse(mock_count_words.called) @mock.patch.object(Stattext, '_count_lines') def test_total_lines(self, mock_count_lines): self.stattext.total_lines = None self.stattext.get_total_lines("fake") mock_count_lines.assert_called_with("fake") @mock.patch.object(Stattext, '_count_lines') def test_count_lines_no_called(self, mock_count_lines): self.stattext.total_lines = 21 self.stattext.get_total_lines("fake") self.assertFalse(mock_count_lines.called) @mock.patch.object(Stattext, '_count_letters') def test_total_letters(self, mock_count_letters): self.stattext.total_letters = None self.stattext.get_total_letters("fake") mock_count_letters.assert_called_with("fake") @mock.patch.object(Stattext, '_count_letters') def test_count_letters_no_called(self, mock_count_letters): self.stattext.total_letters = 21 self.stattext.get_total_letters("fake") self.assertFalse(mock_count_letters.called) @mock.patch.object(Stattext, '_avg_letters') def test_avg_letters_per_word(self, mock_avg_letters): self.stattext.avg_letters_per_word = None self.stattext.get_avg_letters_per_word("fake") mock_avg_letters.assert_called_with("fake", precision=1) @mock.patch.object(Stattext, '_avg_letters') def test_avg_letters_no_called(self, mock_avg_letters): self.stattext.avg_letters_per_word = 21 self.stattext.get_avg_letters_per_word("fake") self.assertFalse(mock_avg_letters.called)