test_metrics.py

#!/usr/bin/python3.10
# -*-coding:Utf-8 -*

#==============================================================================
# test_metrics: Metrics Testing Module
#------------------------------------------------------------------------------
# Contains tests for verifying functionality of the Metrics class.
#==============================================================================

import unittest
import os
from context import ontoScorer
from ontoScorer.ontology import Ontology
from ontoScorer.metrics import Metrics

class TestMetrics(unittest.TestCase):

    def setUp(self):
        DATA_FOLDER_PATH = f'{os.path.dirname(os.path.abspath(__file__))}/test_data'
        self.ontology1_path = f"{DATA_FOLDER_PATH}/ontology_a.ttl"
        self.ontology2_path = f"{DATA_FOLDER_PATH}/ontology_b.ttl"
        self.onto1 = Ontology(self.ontology1_path)
        self.onto2 = Ontology(self.ontology2_path)
        self.metrics = Metrics()

    def test_calculate_scores(self):
        self.metrics.calculate(self.onto1, self.onto2)
        for key in self.metrics.scores:
            self.assertTrue(0 <= self.metrics.scores[key]["precision"] <= 1)
            self.assertTrue(0 <= self.metrics.scores[key]["recall"] <= 1)
            self.assertTrue(0 <= self.metrics.scores[key]["f1"] <= 1)


    def test_print_scores(self):
        self.metrics.calculate(self.onto1, self.onto2)
        self.metrics.print_scores()


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