Ejemplo n.º 1
0
 def test_mention_type(self):
     self.assertEqual(
         "NAM",
         Mention.from_document(
             Span(37, 37),
             self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NAM",
         Mention.from_document(
             Span(11, 12),
             self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NAM",
         Mention.from_document(
             Span(31, 32),
             self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NAM",
         Mention.from_document(
             Span(8, 8),
             self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NOM",
         Mention.from_document(
             Span(33, 34),
             self.real_document).attributes["type"])
Ejemplo n.º 2
0
 def test_mention_get_context(self):
     self.assertEqual(
         ["laid", "plans"], Mention.from_document(Span(21, 27), self.date_mention_document).get_context(2)
     )
     self.assertEqual(None, Mention.from_document(Span(21, 27), self.date_mention_document).get_context(0))
     self.assertEqual(["through"], Mention.from_document(Span(31, 32), self.date_mention_document).get_context(-1))
     self.assertEqual(None, Mention.from_document(Span(21, 27), self.date_mention_document).get_context(1000))
Ejemplo n.º 3
0
 def test_mention_set_id(self):
     self.assertEqual(0, Mention.from_document(Span(33, 34), self.real_document).attributes["annotated_set_id"])
     self.assertEqual(
         4, Mention.from_document(Span(6, 10), self.complicated_mention_document).attributes["annotated_set_id"]
     )
     self.assertEqual(
         1, Mention.from_document(Span(3, 3), self.complicated_mention_document).attributes["annotated_set_id"]
     )
Ejemplo n.º 4
0
    def test_mention_tokens(self):
        expected = ["the", "massacre"]
        self.assertEqual(expected, Mention.from_document(Span(33, 34), self.real_document).attributes["tokens"])

        expected = "the massacre"
        self.assertEqual(
            expected, Mention.from_document(Span(33, 34), self.real_document).attributes["tokens_as_lowercase_string"]
        )
Ejemplo n.º 5
0
 def test_mention_get_sentence_id(self):
     self.assertEqual(
         0,
         Mention.from_document(Span(
             13, 20), self.real_document).attributes["sentence_id"])
     self.assertEqual(
         1,
         Mention.from_document(Span(
             33, 34), self.real_document).attributes["sentence_id"])
Ejemplo n.º 6
0
 def test_mention_get_fine_type(self):
     self.assertEqual("DEF", Mention.from_document(Span(33, 34), self.real_document).attributes["fine_type"])
     self.assertEqual("DEF", Mention.from_document(Span(21, 27), self.date_mention_document).attributes["fine_type"])
     self.assertEqual(
         "INDEF", Mention.from_document(Span(22, 22), self.date_mention_document).attributes["fine_type"]
     )
     self.assertEqual(
         "POSS_ADJ", Mention.from_document(Span(45, 45), self.date_mention_document).attributes["fine_type"]
     )
Ejemplo n.º 7
0
    def test_mention_get_head(self):
        expected = ["massacre"]
        self.assertEqual(expected, Mention.from_document(Span(33, 34), self.real_document).attributes["head"])

        expected = ["Wedding"]
        self.assertEqual(expected, Mention.from_document(Span(3, 3), self.for_head_document).attributes["head"])

        expected = "wedding"
        self.assertEqual(
            expected, Mention.from_document(Span(3, 3), self.for_head_document).attributes["head_as_lowercase_string"]
        )
Ejemplo n.º 8
0
 def test_mention_get_sentence_id(self):
     self.assertEqual(
         0,
         Mention.from_document(
             Span(13, 20),
             self.real_document).attributes["sentence_id"])
     self.assertEqual(
         1,
         Mention.from_document(
             Span(33, 34),
             self.real_document).attributes["sentence_id"])
Ejemplo n.º 9
0
    def test_mention_get_ancestry(self):
        expected = "-L-VBN-L-NONE"
        self.assertEqual(
            expected,
            Mention.from_document(Span(11, 11),
                                  self.real_document).attributes["ancestry"])

        expected = "-R-NNS-R-VBN"
        self.assertEqual(
            expected,
            Mention.from_document(Span(0, 0),
                                  self.real_document).attributes["ancestry"])
Ejemplo n.º 10
0
    def test_mention_tokens(self):
        expected = ["the", "massacre"]
        self.assertEqual(
            expected,
            Mention.from_document(Span(33, 34),
                                  self.real_document).attributes["tokens"])

        expected = "the massacre"
        self.assertEqual(
            expected,
            Mention.from_document(
                Span(33, 34),
                self.real_document).attributes["tokens_as_lowercase_string"])
Ejemplo n.º 11
0
 def test_mention_set_id(self):
     self.assertEqual(
         0,
         Mention.from_document(Span(
             33, 34), self.real_document).attributes["annotated_set_id"])
     self.assertEqual(
         4,
         Mention.from_document(Span(6, 10),
                               self.complicated_mention_document).
         attributes["annotated_set_id"])
     self.assertEqual(
         1,
         Mention.from_document(Span(3, 3), self.complicated_mention_document
                               ).attributes["annotated_set_id"])
Ejemplo n.º 12
0
 def test_mention_get_fine_type(self):
     self.assertEqual(
         "DEF",
         Mention.from_document(Span(33, 34),
                               self.real_document).attributes["fine_type"])
     self.assertEqual(
         "DEF",
         Mention.from_document(Span(
             21, 27), self.date_mention_document).attributes["fine_type"])
     self.assertEqual(
         "INDEF",
         Mention.from_document(Span(
             22, 22), self.date_mention_document).attributes["fine_type"])
     self.assertEqual(
         "POSS_ADJ",
         Mention.from_document(Span(
             45, 45), self.date_mention_document).attributes["fine_type"])
Ejemplo n.º 13
0
 def test_mention_get_context(self):
     self.assertEqual(
         ["laid", "plans"],
         Mention.from_document(Span(21, 27),
                               self.date_mention_document).get_context(2))
     self.assertEqual(
         None,
         Mention.from_document(Span(21, 27),
                               self.date_mention_document).get_context(0))
     self.assertEqual(
         ["through"],
         Mention.from_document(Span(31, 32),
                               self.date_mention_document).get_context(-1))
     self.assertEqual(
         None,
         Mention.from_document(Span(
             21, 27), self.date_mention_document).get_context(1000))
Ejemplo n.º 14
0
    def test_mention_get_head(self):
        expected = ["massacre"]
        self.assertEqual(
            expected,
            Mention.from_document(Span(33, 34),
                                  self.real_document).attributes["head"])

        expected = ["Wedding"]
        self.assertEqual(
            expected,
            Mention.from_document(Span(3, 3),
                                  self.for_head_document).attributes["head"])

        expected = "wedding"
        self.assertEqual(
            expected,
            Mention.from_document(
                Span(3, 3),
                self.for_head_document).attributes["head_as_lowercase_string"])
Ejemplo n.º 15
0
 def test_mention_type(self):
     self.assertEqual(
         "NAM",
         Mention.from_document(Span(
             37, 37), self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NAM",
         Mention.from_document(Span(
             11, 12), self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NAM",
         Mention.from_document(Span(
             31, 32), self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NAM",
         Mention.from_document(Span(
             8, 8), self.date_mention_document).attributes["type"])
     self.assertEqual(
         "NOM",
         Mention.from_document(Span(33, 34),
                               self.real_document).attributes["type"])
Ejemplo n.º 16
0
 def test_mention_get_governor(self):
     expected = "massacred"
     self.assertEqual(
         expected,
         Mention.from_document(Span(0, 1),
                               self.real_document).attributes["governor"])
Ejemplo n.º 17
0
 def test_mention_get_head_span(self):
     self.assertEqual(
         Span(9, 10), Mention.from_document(Span(8, 10), self.for_head_document).attributes["head_span"]
     )
Ejemplo n.º 18
0
    def test_mention_get_ancestry(self):
        expected = "-L-VBN-L-NONE"
        self.assertEqual(expected, Mention.from_document(Span(11, 11), self.real_document).attributes["ancestry"])

        expected = "-R-NNS-R-VBN"
        self.assertEqual(expected, Mention.from_document(Span(0, 0), self.real_document).attributes["ancestry"])
Ejemplo n.º 19
0
 def test_mention_get_governor(self):
     expected = "massacred"
     self.assertEqual(expected, Mention.from_document(Span(0, 1), self.real_document).attributes["governor"])
Ejemplo n.º 20
0
 def test_mention_get_head_span(self):
     self.assertEqual(
         Span(9, 10),
         Mention.from_document(Span(
             8, 10), self.for_head_document).attributes["head_span"])