bugfix

new-village · Aug 12, 2024 · 5a0aee9 · 5a0aee9
1 parent 00c5807
commit 5a0aee9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_enrich.py b/test/test_enrich.py
@@ -22,7 +22,7 @@ def test_enrich_kind(self):
         """Test the enrich_kind function to ensure it correctly maps 'kind' to 'std_legal_entity'."""
         result = enrich_kind(self.df.copy())
         self.assertIn('std_legal_entity', result.columns)
-        expected_entities = ['National Agency', 'K.K.', 'K.K.', 'K.K.', 'Y.K.']
+        expected_entities = ['国の機関', '株式会社', '株式会社', '株式会社', '有限会社']
         for i, entity in enumerate(expected_entities):
             self.assertEqual(result.iloc[i]['std_legal_entity'], entity)