diff options
Diffstat (limited to 'test/extractor/text/test_metrics.py')
| -rw-r--r-- | test/extractor/text/test_metrics.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/extractor/text/test_metrics.py b/test/extractor/text/test_metrics.py index 9cc6a94..6d87889 100644 --- a/test/extractor/text/test_metrics.py +++ b/test/extractor/text/test_metrics.py @@ -55,8 +55,8 @@ class TestTextMetrics(unittest.TestCase): path = os.path.join(os.path.dirname(__file__), 'example-en.txt') # fetch document text = rdr(path) + # extracts all specified predicates triples = set(ext.extract(subject, text, principals)) - self.assertSetEqual({(s,p,o) for s,p,o in triples if p.uri != ns.bse.vocabulary_entropy}, { (subject, ext.schema.predicate(ns.bse.num_characters), 21997), (subject, ext.schema.predicate(ns.bse.num_paragraphs), 48), @@ -66,6 +66,9 @@ class TestTextMetrics(unittest.TestCase): entropy = {o for s,p,o in triples if p.uri == ns.bse.vocabulary_entropy} self.assertEqual(len(entropy), 1) self.assertAlmostEqual(list(entropy)[0], 8.830360505) + # skip unknown predicates + self.assertSetEqual(set(), set(ext.extract(subject, text, + {ext.schema.predicate(ns.bsfs.Predicate).child(ns.bse.unknown)}))) ## main ## |