Skip to content
This repository was archived by the owner on Sep 9, 2026. It is now read-only.

Commit 6bb36d9

Browse files
AnneYang720AnneYang720
committed
refactor: rename store to index
Signed-off-by: AnneY <evangeline-lun@foxmail.com>
1 parent f37f683 commit 6bb36d9

21 files changed

Lines changed: 848 additions & 865 deletions

‎tests/index/base_classes/test_base_doc_store.py‎

Lines changed: 129 additions & 129 deletions
Large diffs are not rendered by default.

‎tests/index/base_classes/test_configs.py‎

Lines changed: 26 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -63,50 +63,50 @@ def python_type_to_db_type(self, x):
6363

6464

6565
def test_defaults():
66-
store = DummyDocIndex[SimpleDoc]()
67-
assert store._db_config.other == 5
68-
assert store._db_config.work_dir == '.'
69-
assert store._runtime_config.default_column_config[str] == {
66+
index = DummyDocIndex[SimpleDoc]()
67+
assert index._db_config.other == 5
68+
assert index._db_config.work_dir == '.'
69+
assert index._runtime_config.default_column_config[str] == {
7070
'dim': 128,
7171
'space': 'l2',
7272
}
7373

7474

7575
def test_set_by_class():
7676
# change all settings
77-
store = DummyDocIndex[SimpleDoc](DBConfig(work_dir='hi', other=10))
78-
assert store._db_config.other == 10
79-
assert store._db_config.work_dir == 'hi'
80-
store.configure(RuntimeConfig(default_column_config={}, default_ef=10))
81-
assert store._runtime_config.default_column_config == {}
77+
index = DummyDocIndex[SimpleDoc](DBConfig(work_dir='hi', other=10))
78+
assert index._db_config.other == 10
79+
assert index._db_config.work_dir == 'hi'
80+
index.configure(RuntimeConfig(default_column_config={}, default_ef=10))
81+
assert index._runtime_config.default_column_config == {}
8282

8383
# change only some settings
84-
store = DummyDocIndex[SimpleDoc](DBConfig(work_dir='hi'))
85-
assert store._db_config.other == 5
86-
assert store._db_config.work_dir == 'hi'
87-
store.configure(RuntimeConfig(default_column_config={}))
88-
assert store._runtime_config.default_column_config == {}
84+
index = DummyDocIndex[SimpleDoc](DBConfig(work_dir='hi'))
85+
assert index._db_config.other == 5
86+
assert index._db_config.work_dir == 'hi'
87+
index.configure(RuntimeConfig(default_column_config={}))
88+
assert index._runtime_config.default_column_config == {}
8989

9090

9191
def test_set_by_kwargs():
9292
# change all settings
93-
store = DummyDocIndex[SimpleDoc](work_dir='hi', other=10)
94-
assert store._db_config.other == 10
95-
assert store._db_config.work_dir == 'hi'
96-
store.configure(default_column_config={}, default_ef=10)
97-
assert store._runtime_config.default_column_config == {}
93+
index = DummyDocIndex[SimpleDoc](work_dir='hi', other=10)
94+
assert index._db_config.other == 10
95+
assert index._db_config.work_dir == 'hi'
96+
index.configure(default_column_config={}, default_ef=10)
97+
assert index._runtime_config.default_column_config == {}
9898

9999
# change only some settings
100-
store = DummyDocIndex[SimpleDoc](work_dir='hi')
101-
assert store._db_config.other == 5
102-
assert store._db_config.work_dir == 'hi'
103-
store.configure(default_column_config={})
104-
assert store._runtime_config.default_column_config == {}
100+
index = DummyDocIndex[SimpleDoc](work_dir='hi')
101+
assert index._db_config.other == 5
102+
assert index._db_config.work_dir == 'hi'
103+
index.configure(default_column_config={})
104+
assert index._runtime_config.default_column_config == {}
105105

106106

107107
def test_default_column_config():
108-
store = DummyDocIndex[SimpleDoc]()
109-
assert store._runtime_config.default_column_config == {
108+
index = DummyDocIndex[SimpleDoc]()
109+
assert index._runtime_config.default_column_config == {
110110
str: {
111111
'dim': 128,
112112
'space': 'l2',

‎tests/index/elastic/v7/test_column_config.py‎

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -13,20 +13,20 @@ class MyDoc(BaseDoc):
1313
text: str
1414
color: str = Field(col_type='keyword')
1515

16-
store = ElasticV7DocIndex[MyDoc]()
16+
index = ElasticV7DocIndex[MyDoc]()
1717
index_docs = [
1818
MyDoc(id='0', text='hello world', color='red'),
1919
MyDoc(id='1', text='never gonna give you up', color='blue'),
2020
MyDoc(id='2', text='we are the world', color='green'),
2121
]
22-
store.index(index_docs)
22+
index.index(index_docs)
2323

2424
query = 'world'
25-
docs, _ = store.text_search(query, search_field='text')
25+
docs, _ = index.text_search(query, search_field='text')
2626
assert [doc.id for doc in docs] == ['0', '2']
2727

2828
filter_query = {'terms': {'color': ['red', 'blue']}}
29-
docs = store.filter(filter_query)
29+
docs = index.filter(filter_query)
3030
assert [doc.id for doc in docs] == ['0', '1']
3131

3232

@@ -44,33 +44,33 @@ class MyDoc(BaseDoc):
4444
}
4545
)
4646

47-
store = ElasticV7DocIndex[MyDoc]()
47+
index = ElasticV7DocIndex[MyDoc]()
4848
doc = [
4949
MyDoc(manager={'age': 25, 'name': {'first': 'Rachel', 'last': 'Green'}}),
5050
MyDoc(manager={'age': 30, 'name': {'first': 'Monica', 'last': 'Geller'}}),
5151
MyDoc(manager={'age': 35, 'name': {'first': 'Phoebe', 'last': 'Buffay'}}),
5252
]
53-
store.index(doc)
53+
index.index(doc)
5454
id_ = doc[0].id
55-
assert store[id_].id == id_
56-
assert store[id_].manager == doc[0].manager
55+
assert index[id_].id == id_
56+
assert index[id_].manager == doc[0].manager
5757

5858
filter_query = {'range': {'manager.age': {'gte': 30}}}
59-
docs = store.filter(filter_query)
59+
docs = index.filter(filter_query)
6060
assert [doc.id for doc in docs] == [doc[1].id, doc[2].id]
6161

6262

6363
def test_field_geo_point():
6464
class MyDoc(BaseDoc):
6565
location: dict = Field(col_type='geo_point')
6666

67-
store = ElasticV7DocIndex[MyDoc]()
67+
index = ElasticV7DocIndex[MyDoc]()
6868
doc = [
6969
MyDoc(location={'lat': 40.12, 'lon': -72.34}),
7070
MyDoc(location={'lat': 41.12, 'lon': -73.34}),
7171
MyDoc(location={'lat': 42.12, 'lon': -74.34}),
7272
]
73-
store.index(doc)
73+
index.index(doc)
7474

7575
query = {
7676
'query': {
@@ -83,7 +83,7 @@ class MyDoc(BaseDoc):
8383
},
8484
}
8585

86-
docs, _ = store.execute_query(query)
86+
docs, _ = index.execute_query(query)
8787
assert [doc['id'] for doc in docs] == [doc[0].id, doc[1].id]
8888

8989

@@ -92,7 +92,7 @@ class MyDoc(BaseDoc):
9292
expected_attendees: dict = Field(col_type='integer_range')
9393
time_frame: dict = Field(col_type='date_range', format='yyyy-MM-dd')
9494

95-
store = ElasticV7DocIndex[MyDoc]()
95+
index = ElasticV7DocIndex[MyDoc]()
9696
doc = [
9797
MyDoc(
9898
expected_attendees={'gte': 10, 'lt': 20},
@@ -107,7 +107,7 @@ class MyDoc(BaseDoc):
107107
time_frame={'gte': '2023-03-01', 'lt': '2023-04-01'},
108108
),
109109
]
110-
store.index(doc)
110+
index.index(doc)
111111

112112
query = {
113113
'query': {
@@ -127,5 +127,5 @@ class MyDoc(BaseDoc):
127127
}
128128
},
129129
}
130-
docs, _ = store.execute_query(query)
130+
docs, _ = index.execute_query(query)
131131
assert [doc['id'] for doc in docs] == [doc[0].id, doc[1].id]

0 commit comments

Comments
 (0)