Browse Source

将test_embed_loader的almost equal范围稍稍调大一些,以防止偶然的测试不通过

tags/v0.4.10
xuyige 5 years ago
parent
commit
bedb792b3c
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      test/io/test_embed_loader.py

+ 3
- 3
test/io/test_embed_loader.py View File

@@ -16,7 +16,7 @@ class TestEmbedLoader(unittest.TestCase):
self.assertEqual(g_m.shape, (4, 50))
w_m = EmbedLoader.load_with_vocab(word2vec, vocab, normalize=True)
self.assertEqual(w_m.shape, (4, 50))
self.assertAlmostEqual(np.linalg.norm(w_m, axis=1).sum(), 4)
self.assertAlmostEqual(np.linalg.norm(w_m, axis=1).sum(), delta=1e-3)
def test_load_without_vocab(self):
words = ['the', 'of', 'in', 'a', 'to', 'and']
@@ -28,13 +28,13 @@ class TestEmbedLoader(unittest.TestCase):
self.assertIn(word, vocab)
w_m, vocab = EmbedLoader.load_without_vocab(word2vec, normalize=True)
self.assertEqual(w_m.shape, (8, 50))
self.assertAlmostEqual(np.linalg.norm(w_m, axis=1).sum(), 8)
self.assertAlmostEqual(np.linalg.norm(w_m, axis=1).sum(), delta=1e-4)
for word in words:
self.assertIn(word, vocab)
# no unk
w_m, vocab = EmbedLoader.load_without_vocab(word2vec, normalize=True, unknown=None)
self.assertEqual(w_m.shape, (7, 50))
self.assertAlmostEqual(np.linalg.norm(w_m, axis=1).sum(), 7)
self.assertAlmostEqual(np.linalg.norm(w_m, axis=1).sum(), delta=1e-4)
for word in words:
self.assertIn(word, vocab)


Loading…
Cancel
Save