tests/test_metrics.py

import unittest

from gtrox.metrics import calculate_metrics, count_fillers, count_repetitions


class MetricsTests(unittest.TestCase):
    def test_counts_multiword_fillers_and_repetitions(self):
        transcript = "Um, I I actually like this, you know."

        self.assertEqual(count_fillers(transcript), 4)
        self.assertEqual(count_repetitions(transcript), 1)

    def test_calculates_duration_based_metrics(self):
        metrics = calculate_metrics("One two three four", duration_seconds=2)

        self.assertEqual(metrics["total_words"], 4)
        self.assertEqual(metrics["wpm"], 120.0)


if __name__ == "__main__":
    unittest.main()