From 5a0aee99c2519d439625c20b8481daccb3d1fdba Mon Sep 17 00:00:00 2001 From: new-village Date: Mon, 12 Aug 2024 13:58:59 +0000 Subject: [PATCH] bugfix --- test/test_enrich.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_enrich.py b/test/test_enrich.py index 4fc6ef8..1f3caa2 100644 --- 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)