Skip to content

Commit f26677a

Browse files
committed
Fixed 'test_yuanrong_storage_client_e2e.py' about strategy_tag
Signed-off-by: dpj135 <958208521@qq.com>
1 parent 23fd669 commit f26677a

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

tests/test_yuanrong_storage_client_e2e.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,8 @@ def test_cpu_only_flow(self, config):
167167

168168
# Put & Verify Meta
169169
meta = client.put(keys, vals)
170-
assert all(m == "KVClient" for m in meta)
170+
# b"\x01" is a tag added by YuanrongStorageClient, indicating that it is processed via General KV path.
171+
assert all(m == b"\x02" for m in meta)
171172

172173
# Get & Verify Values
173174
ret = client.get(keys, shp, dt, meta)
@@ -186,7 +187,8 @@ def test_npu_only_flow(self, config):
186187
client = self.client_cls(config)
187188

188189
meta = client.put(keys, vals)
189-
assert all(m == "DsTensorClient" for m in meta)
190+
# b"\x01" is a tag added by YuanrongStorageClient, indicating that it is processed via NPU path.
191+
assert all(m == b"\x01" for m in meta)
190192

191193
ret = client.get(keys, shp, dt, meta)
192194
for o, r in zip(vals, ret, strict=True):
@@ -199,7 +201,7 @@ def test_mixed_flow(self, config):
199201
client = self.client_cls(config)
200202

201203
meta = client.put(keys, vals)
202-
assert set(meta) == {"DsTensorClient", "KVClient"}
204+
assert set(meta) == {b"\x01", b"\x02"}
203205

204206
ret = client.get(keys, shp, dt, meta)
205207
for o, r in zip(vals, ret, strict=True):

0 commit comments

Comments
 (0)