fix issues

ikuyamada · ikuyamada · commit af8c62cfa319 · 2024-01-06T17:18:13.000+09:00
diff --git a/tests/test_dictionary.py b/tests/test_dictionary.py
@@ -257,7 +257,7 @@ def validate(obj):
             s2 = obj.serialize()
             for key in s1.keys():
                 if isinstance(s1[key], np.ndarray):
-                    np.array_equal(s1[key], s2[key])
+                    self.assertTrue(np.array_equal(s1[key], s2[key]))
                 else:
                     self.assertEqual(s1[key], s2[key])
 
diff --git a/tests/test_link_graph.py b/tests/test_link_graph.py
@@ -82,7 +82,7 @@ def validate(obj):
             s2 = obj.serialize()
             for key in s1.keys():
                 if isinstance(s1[key], np.ndarray):
-                    np.array_equal(s1[key], s2[key])
+                    self.assertTrue(np.array_equal(s1[key], s2[key]))
                 else:
                     self.assertEqual(s1[key], s2[key])
 
diff --git a/wikipedia2vec/__init__.py b/wikipedia2vec/__init__.py
@@ -1,5 +1,2 @@
-# -*- coding: utf-8 -*-
-# License: Apache License 2.0
-
 from .dictionary import Dictionary
 from .wikipedia2vec import Wikipedia2Vec
