demand-agent-new/examples/demand/tests/test_pg_pattern_repository.py

import unittest

from examples.demand.pg_pattern_repository import _is_element_type_dimension


class PgPatternRepositoryTest(unittest.TestCase):
    def test_topic_itemset_need_dimension_does_not_filter_element_type(self):
        self.assertFalse(_is_element_type_dimension("需求"))

    def test_real_element_dimensions_filter_element_type(self):
        self.assertTrue(_is_element_type_dimension("实质"))
        self.assertTrue(_is_element_type_dimension("形式"))
        self.assertTrue(_is_element_type_dimension("意图"))


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