From f22991698ad02a459e0a22ef3f486e29fc112f72 Mon Sep 17 00:00:00 2001 From: Yige Xu Date: Wed, 11 Sep 2019 02:19:27 +0800 Subject: [PATCH] add assert in test_elmo_embedding --- test/embeddings/test_elmo_embedding.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/embeddings/test_elmo_embedding.py b/test/embeddings/test_elmo_embedding.py index bfb31659..ed6910b4 100644 --- a/test/embeddings/test_elmo_embedding.py +++ b/test/embeddings/test_elmo_embedding.py @@ -25,6 +25,7 @@ class TestRunElmo(unittest.TestCase): words = torch.LongTensor([[0, 1, 2]]) hidden = elmo_embed(words) print(hidden.size()) + self.assertEqual(hidden.size(), (1, 3, elmo_embed.embedding_dim)) def test_elmo_embedding_layer_assertion(self): vocab = Vocabulary().add_word_lst("This is a test .".split())