From c64322fc9dad8fb6c4fb484a1899404f1b87c7fa Mon Sep 17 00:00:00 2001 From: Eric Joanis Date: Mon, 1 May 2023 12:44:55 -0400 Subject: [PATCH] test: execise api/v1/g2p with and without tokenize --- g2p/tests/test_api_resources.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) diff --git a/g2p/tests/test_api_resources.py b/g2p/tests/test_api_resources.py index 79e091f5..2e35ba84 100755 --- a/g2p/tests/test_api_resources.py +++ b/g2p/tests/test_api_resources.py @@ -143,6 +143,38 @@ def test_g2p_conversion(self): ) self.assertEqual(missing_response.status_code, 404) + def test_g2p_conversion_with_tok(self): + params_with_tok = { + "in-lang": "fra", + "out-lang": "eng-arpabet", + "text": "ceci, celà", + "debugger": True, + "index": True, + "tokenize": True, + } + response = self.client().get( + self.conversion_route, query_string=params_with_tok + ) + self.assertEqual(response.status_code, 200) + res_json_tok = response.get_json() + self.assertEqual(res_json_tok["debugger"][0][0][0]["input"], "ceci") + + params_no_tok = { + "in-lang": "fra", + "out-lang": "eng-arpabet", + "text": "ceci, celà", + "debugger": True, + "index": True, + "tokenize": False, + } + response = self.client().get(self.conversion_route, query_string=params_no_tok) + self.assertEqual(response.status_code, 200) + res_json_no_tok = response.get_json() + self.assertNotEqual(res_json_tok, res_json_no_tok) + self.assertEqual(res_json_no_tok["debugger"][0][0][0]["input"], "ceci, celà") + + self.assertNotEqual(res_json_tok["debugger"], res_json_no_tok["debugger"]) + if __name__ == "__main__": main()