From 49d8a36c0ea0f0bf5c0dbefe294f761c70521577 Mon Sep 17 00:00:00 2001 From: medaka <36759068+medaka0213@users.noreply.github.com> Date: Fri, 20 Sep 2024 21:24:37 +0900 Subject: [PATCH] =?UTF-8?q?=E3=83=86=E3=82=B9=E3=83=88=E5=8A=B9=E7=8E=87?= =?UTF-8?q?=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_search_table.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/test_search_table.py b/tests/test_search_table.py index 71c7ae3..2e4fd8b 100644 --- a/tests/test_search_table.py +++ b/tests/test_search_table.py @@ -42,7 +42,7 @@ class UserNotFound(BaseModel): class TestSearchTable(unittest.TestCase): def setUp(self): - for i in range(500): + for i in range(200): test = User( name=f"test{str(i).zfill(3)}", age=i, @@ -118,7 +118,7 @@ def test_search_staged_filter(self): searchEx = [{ "FilterStatus": util_b.FilterStatus.STAGED, "IndexName": table.__search_index__, - "KeyConditionExpression": Key("sk").eq("search_user_name") & Key("data").begins_with("test2"), + "KeyConditionExpression": Key("sk").eq("search_user_name") & Key("data").begins_with("test1"), }, { "FilterStatus": util_b.FilterStatus.FILTER, "FilterMethod": util_b.attr_method("description", "odd", util_b.QueryType.CONTAINS), @@ -144,30 +144,30 @@ def test_search_filter(self): "FilterExpression": Attr("description").contains("odd"), }] res = table.search("user", *searchEx) - self.assertEqual(len(res), 250) + self.assertEqual(len(res), 100) for r in res: self.assertIn("odd", r["description"]) # pk_only res = table.search("user", *searchEx, pk_only=True) - self.assertEqual(len(res), 250) + self.assertEqual(len(res), 100) for r in res: self.assertIsInstance(r, str) def test_search_all(self): """全件検索""" res = table.search("user") - self.assertEqual(len(res), 500) + self.assertEqual(len(res), 200) # pk_only res = table.search("user", pk_only=True) - self.assertEqual(len(res), 500) + self.assertEqual(len(res), 200) for r in res: self.assertIsInstance(r, str) def test_all_items(self): res = table.all_items() - self.assertEqual(len(res), 501) + self.assertEqual(len(res), 201) def test_list_models(self): res = table.list_models()