test embed_sentences used with a string

yannvgn · Dec 18, 2019 · 4048636 · 4048636
1 parent dac8cfc
commit 4048636
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_laser.py b/tests/test_laser.py
@@ -20,6 +20,8 @@ def test_laser():
             lang='en').shape == (2, 1024)
         assert laser.embed_sentences(['hello world!', "j'aime les pâtes"],
                                      lang=['en', 'fr']).shape == (2, 1024)
+        assert laser.embed_sentences('hello world!',
+                                     lang='en').shape == (1, 1024)
 
 
 def test_similarity(test_data):