mirror of
https://github.com/deepmodeling/Uni-Lab-OS
synced 2026-03-31 18:43:05 +00:00
539 lines
19 KiB
Python
539 lines
19 KiB
Python
import pytest
|
||
import json
|
||
import os
|
||
import asyncio
|
||
import collections
|
||
from typing import List, Dict, Any
|
||
|
||
from pylabrobot.resources import Coordinate
|
||
from pylabrobot.resources.opentrons.tip_racks import opentrons_96_tiprack_300ul, opentrons_96_tiprack_10ul
|
||
from pylabrobot.resources.opentrons.plates import corning_96_wellplate_360ul_flat, nest_96_wellplate_2ml_deep
|
||
|
||
from unilabos.devices.liquid_handling.prcxi.prcxi import (
|
||
PRCXI9300Deck,
|
||
PRCXI9300Container,
|
||
PRCXI9300Trash,
|
||
PRCXI9300Handler,
|
||
PRCXI9300Backend,
|
||
DefaultLayout,
|
||
Material,
|
||
WorkTablets,
|
||
MatrixInfo
|
||
)
|
||
|
||
|
||
@pytest.fixture
|
||
def prcxi_materials() -> Dict[str, Any]:
|
||
"""加载 PRCXI 物料数据"""
|
||
print("加载 PRCXI 物料数据...")
|
||
material_path = os.path.join(os.path.dirname(__file__), "..", "..", "unilabos", "devices", "liquid_handling", "prcxi", "prcxi_material.json")
|
||
with open(material_path, "r", encoding="utf-8") as f:
|
||
data = json.load(f)
|
||
print(f"加载了 {len(data)} 条物料数据")
|
||
return data
|
||
|
||
|
||
@pytest.fixture
|
||
def prcxi_9300_deck() -> PRCXI9300Deck:
|
||
"""创建 PRCXI 9300 工作台"""
|
||
return PRCXI9300Deck(name="PRCXI_Deck_9300", size_x=100, size_y=100, size_z=100, model="9300")
|
||
|
||
|
||
@pytest.fixture
|
||
def prcxi_9320_deck() -> PRCXI9300Deck:
|
||
"""创建 PRCXI 9320 工作台"""
|
||
return PRCXI9300Deck(name="PRCXI_Deck_9320", size_x=100, size_y=100, size_z=100, model="9320")
|
||
|
||
|
||
@pytest.fixture
|
||
def prcxi_9300_handler(prcxi_9300_deck) -> PRCXI9300Handler:
|
||
"""创建 PRCXI 9300 处理器（模拟模式）"""
|
||
return PRCXI9300Handler(
|
||
deck=prcxi_9300_deck,
|
||
host="192.168.1.201",
|
||
port=9999,
|
||
timeout=10.0,
|
||
channel_num=8,
|
||
axis="Left",
|
||
setup=False,
|
||
debug=True,
|
||
simulator=True,
|
||
matrix_id="test-matrix-9300"
|
||
)
|
||
|
||
|
||
@pytest.fixture
|
||
def prcxi_9320_handler(prcxi_9320_deck) -> PRCXI9300Handler:
|
||
"""创建 PRCXI 9320 处理器（模拟模式）"""
|
||
return PRCXI9300Handler(
|
||
deck=prcxi_9320_deck,
|
||
host="192.168.1.201",
|
||
port=9999,
|
||
timeout=10.0,
|
||
channel_num=1,
|
||
axis="Right",
|
||
setup=False,
|
||
debug=True,
|
||
simulator=True,
|
||
matrix_id="test-matrix-9320",
|
||
is_9320=True
|
||
)
|
||
|
||
|
||
@pytest.fixture
|
||
def tip_rack_300ul(prcxi_materials) -> PRCXI9300Container:
|
||
"""创建 300μL 枪头盒"""
|
||
tip_rack = PRCXI9300Container(
|
||
name="tip_rack_300ul",
|
||
size_x=50,
|
||
size_y=50,
|
||
size_z=10,
|
||
category="tip_rack",
|
||
ordering=collections.OrderedDict()
|
||
)
|
||
tip_rack.load_state({
|
||
"Material": {
|
||
"uuid": prcxi_materials["300μL Tip头"]["uuid"],
|
||
"Code": "ZX-001-300",
|
||
"Name": "300μL Tip头"
|
||
}
|
||
})
|
||
return tip_rack
|
||
|
||
|
||
@pytest.fixture
|
||
def tip_rack_10ul(prcxi_materials) -> PRCXI9300Container:
|
||
"""创建 10μL 枪头盒"""
|
||
tip_rack = PRCXI9300Container(
|
||
name="tip_rack_10ul",
|
||
size_x=50,
|
||
size_y=50,
|
||
size_z=10,
|
||
category="tip_rack",
|
||
ordering=collections.OrderedDict()
|
||
)
|
||
tip_rack.load_state({
|
||
"Material": {
|
||
"uuid": prcxi_materials["10μL加长 Tip头"]["uuid"],
|
||
"Code": "ZX-001-10+",
|
||
"Name": "10μL加长 Tip头"
|
||
}
|
||
})
|
||
return tip_rack
|
||
|
||
|
||
@pytest.fixture
|
||
def well_plate_96(prcxi_materials) -> PRCXI9300Container:
|
||
"""创建 96 孔板"""
|
||
plate = PRCXI9300Container(
|
||
name="well_plate_96",
|
||
size_x=50,
|
||
size_y=50,
|
||
size_z=10,
|
||
category="plate",
|
||
ordering=collections.OrderedDict()
|
||
)
|
||
plate.load_state({
|
||
"Material": {
|
||
"uuid": prcxi_materials["96深孔板"]["uuid"],
|
||
"Code": "ZX-019-2.2",
|
||
"Name": "96深孔板"
|
||
}
|
||
})
|
||
return plate
|
||
|
||
|
||
@pytest.fixture
|
||
def deep_well_plate(prcxi_materials) -> PRCXI9300Container:
|
||
"""创建深孔板"""
|
||
plate = PRCXI9300Container(
|
||
name="deep_well_plate",
|
||
size_x=50,
|
||
size_y=50,
|
||
size_z=10,
|
||
category="plate",
|
||
ordering=collections.OrderedDict()
|
||
)
|
||
plate.load_state({
|
||
"Material": {
|
||
"uuid": prcxi_materials["96深孔板"]["uuid"],
|
||
"Code": "ZX-019-2.2",
|
||
"Name": "96深孔板"
|
||
}
|
||
})
|
||
return plate
|
||
|
||
|
||
@pytest.fixture
|
||
def trash_container(prcxi_materials) -> PRCXI9300Trash:
|
||
"""创建垃圾桶"""
|
||
trash = PRCXI9300Trash(name="trash", size_x=50, size_y=50, size_z=10, category="trash")
|
||
trash.load_state({
|
||
"Material": {
|
||
"uuid": prcxi_materials["废弃槽"]["uuid"]
|
||
}
|
||
})
|
||
return trash
|
||
|
||
|
||
@pytest.fixture
|
||
def default_layout_9300() -> DefaultLayout:
|
||
"""创建 PRCXI 9300 默认布局"""
|
||
return DefaultLayout("PRCXI9300")
|
||
|
||
|
||
@pytest.fixture
|
||
def default_layout_9320() -> DefaultLayout:
|
||
"""创建 PRCXI 9320 默认布局"""
|
||
return DefaultLayout("PRCXI9320")
|
||
|
||
|
||
class TestPRCXIDeckSetup:
|
||
"""测试 PRCXI 工作台设置功能"""
|
||
|
||
def test_prcxi_9300_deck_creation(self, prcxi_9300_deck):
|
||
"""测试 PRCXI 9300 工作台创建"""
|
||
assert prcxi_9300_deck.name == "PRCXI_Deck_9300"
|
||
assert len(prcxi_9300_deck.sites) == 6
|
||
assert prcxi_9300_deck._size_x == 100
|
||
assert prcxi_9300_deck._size_y == 100
|
||
assert prcxi_9300_deck._size_z == 100
|
||
|
||
def test_prcxi_9320_deck_creation(self, prcxi_9320_deck):
|
||
"""测试 PRCXI 9320 工作台创建"""
|
||
assert prcxi_9320_deck.name == "PRCXI_Deck_9320"
|
||
assert len(prcxi_9320_deck.sites) == 16
|
||
assert prcxi_9320_deck._size_x == 100
|
||
assert prcxi_9320_deck._size_y == 100
|
||
assert prcxi_9320_deck._size_z == 100
|
||
|
||
def test_container_assignment(self, prcxi_9300_deck, tip_rack_300ul, well_plate_96, trash_container):
|
||
"""测试容器分配到工作台"""
|
||
# 分配枪头盒
|
||
prcxi_9300_deck.assign_child_resource(tip_rack_300ul, location=Coordinate(0, 0, 0))
|
||
assert tip_rack_300ul in prcxi_9300_deck.children
|
||
|
||
# 分配孔板
|
||
prcxi_9300_deck.assign_child_resource(well_plate_96, location=Coordinate(0, 0, 0))
|
||
assert well_plate_96 in prcxi_9300_deck.children
|
||
|
||
# 分配垃圾桶
|
||
prcxi_9300_deck.assign_child_resource(trash_container, location=Coordinate(0, 0, 0))
|
||
assert trash_container in prcxi_9300_deck.children
|
||
|
||
def test_container_material_loading(self, tip_rack_300ul, well_plate_96, prcxi_materials):
|
||
"""测试容器物料信息加载"""
|
||
# 测试枪头盒物料信息
|
||
tip_material = tip_rack_300ul._unilabos_state["Material"]
|
||
assert tip_material["uuid"] == prcxi_materials["300μL Tip头"]["uuid"]
|
||
assert tip_material["Name"] == "300μL Tip头"
|
||
|
||
# 测试孔板物料信息
|
||
plate_material = well_plate_96._unilabos_state["Material"]
|
||
assert plate_material["uuid"] == prcxi_materials["96深孔板"]["uuid"]
|
||
assert plate_material["Name"] == "96深孔板"
|
||
|
||
|
||
class TestPRCXISingleStepOperations:
|
||
"""测试 PRCXI 单步操作功能"""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_pick_up_tips_single_channel(self, prcxi_9320_handler, prcxi_9320_deck, tip_rack_10ul):
|
||
"""测试单通道拾取枪头"""
|
||
# 将枪头盒添加到工作台
|
||
prcxi_9320_deck.assign_child_resource(tip_rack_10ul, location=Coordinate(0, 0, 0))
|
||
|
||
# 初始化处理器
|
||
await prcxi_9320_handler.setup()
|
||
|
||
# 设置枪头盒
|
||
prcxi_9320_handler.set_tiprack([tip_rack_10ul])
|
||
|
||
# 创建模拟的枪头位置
|
||
from pylabrobot.resources import TipSpot, Tip
|
||
tip = Tip(has_filter=False, total_tip_length=10, maximal_volume=10, fitting_depth=5)
|
||
tip_spot = TipSpot("A1", size_x=1, size_y=1, size_z=1, make_tip=lambda: tip)
|
||
tip_rack_10ul.assign_child_resource(tip_spot, location=Coordinate(0, 0, 0))
|
||
|
||
# 直接测试后端方法
|
||
from pylabrobot.liquid_handling import Pickup
|
||
pickup = Pickup(resource=tip_spot, offset=None, tip=tip)
|
||
await prcxi_9320_handler._unilabos_backend.pick_up_tips([pickup], [0])
|
||
|
||
# 验证步骤已添加到待办列表
|
||
assert len(prcxi_9320_handler._unilabos_backend.steps_todo_list) == 1
|
||
step = prcxi_9320_handler._unilabos_backend.steps_todo_list[0]
|
||
assert step["Function"] == "Load"
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_pick_up_tips_multi_channel(self, prcxi_9300_handler, tip_rack_300ul):
|
||
"""测试多通道拾取枪头"""
|
||
# 设置枪头盒
|
||
prcxi_9300_handler.set_tiprack([tip_rack_300ul])
|
||
|
||
# 拾取8个枪头
|
||
tip_spots = tip_rack_300ul.children[:8]
|
||
await prcxi_9300_handler.pick_up_tips(tip_spots, [0, 1, 2, 3, 4, 5, 6, 7])
|
||
|
||
# 验证步骤已添加到待办列表
|
||
assert len(prcxi_9300_handler._unilabos_backend.steps_todo_list) == 1
|
||
step = prcxi_9300_handler._unilabos_backend.steps_todo_list[0]
|
||
assert step["Function"] == "Load"
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_aspirate_single_channel(self, prcxi_9320_handler, well_plate_96):
|
||
"""测试单通道吸取液体"""
|
||
# 设置液体
|
||
well = well_plate_96.get_item("A1")
|
||
prcxi_9320_handler.set_liquid([well], ["water"], [50])
|
||
|
||
# 吸取液体
|
||
await prcxi_9320_handler.aspirate([well], [50], [0])
|
||
|
||
# 验证步骤已添加到待办列表
|
||
assert len(prcxi_9320_handler._unilabos_backend.steps_todo_list) == 1
|
||
step = prcxi_9320_handler._unilabos_backend.steps_todo_list[0]
|
||
assert step["Function"] == "Imbibing"
|
||
assert step["DosageNum"] == 50
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_dispense_single_channel(self, prcxi_9320_handler, well_plate_96):
|
||
"""测试单通道分配液体"""
|
||
# 分配液体
|
||
well = well_plate_96.get_item("A1")
|
||
await prcxi_9320_handler.dispense([well], [25], [0])
|
||
|
||
# 验证步骤已添加到待办列表
|
||
assert len(prcxi_9320_handler._unilabos_backend.steps_todo_list) == 1
|
||
step = prcxi_9320_handler._unilabos_backend.steps_todo_list[0]
|
||
assert step["Function"] == "Tapping"
|
||
assert step["DosageNum"] == 25
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_mix_single_channel(self, prcxi_9320_handler, well_plate_96):
|
||
"""测试单通道混合液体"""
|
||
# 混合液体
|
||
well = well_plate_96.get_item("A1")
|
||
await prcxi_9320_handler.mix([well], mix_time=3, mix_vol=50)
|
||
|
||
# 验证步骤已添加到待办列表
|
||
assert len(prcxi_9320_handler._unilabos_backend.steps_todo_list) == 1
|
||
step = prcxi_9320_handler._unilabos_backend.steps_todo_list[0]
|
||
assert step["Function"] == "Blending"
|
||
assert step["BlendingTimes"] == 3
|
||
assert step["DosageNum"] == 50
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_drop_tips_to_trash(self, prcxi_9320_handler, trash_container):
|
||
"""测试丢弃枪头到垃圾桶"""
|
||
# 丢弃枪头
|
||
await prcxi_9320_handler.drop_tips([trash_container], [0])
|
||
|
||
# 验证步骤已添加到待办列表
|
||
assert len(prcxi_9320_handler._unilabos_backend.steps_todo_list) == 1
|
||
step = prcxi_9320_handler._unilabos_backend.steps_todo_list[0]
|
||
assert step["Function"] == "UnLoad"
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_discard_tips(self, prcxi_9320_handler):
|
||
"""测试丢弃枪头"""
|
||
# 丢弃枪头
|
||
await prcxi_9320_handler.discard_tips([0])
|
||
|
||
# 验证步骤已添加到待办列表
|
||
assert len(prcxi_9320_handler._unilabos_backend.steps_todo_list) == 1
|
||
step = prcxi_9320_handler._unilabos_backend.steps_todo_list[0]
|
||
assert step["Function"] == "UnLoad"
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_liquid_transfer_workflow(self, prcxi_9320_handler, tip_rack_10ul, well_plate_96):
|
||
"""测试完整的液体转移工作流程"""
|
||
# 设置枪头盒和液体
|
||
prcxi_9320_handler.set_tiprack([tip_rack_10ul])
|
||
source_well = well_plate_96.get_item("A1")
|
||
target_well = well_plate_96.get_item("B1")
|
||
prcxi_9320_handler.set_liquid([source_well], ["water"], [100])
|
||
|
||
# 创建协议
|
||
await prcxi_9320_handler.create_protocol(protocol_name="Test Transfer Protocol")
|
||
|
||
# 执行转移流程
|
||
tip_spot = tip_rack_10ul.get_item("A1")
|
||
await prcxi_9320_handler.pick_up_tips([tip_spot], [0])
|
||
await prcxi_9320_handler.aspirate([source_well], [50], [0])
|
||
await prcxi_9320_handler.dispense([target_well], [50], [0])
|
||
await prcxi_9320_handler.discard_tips([0])
|
||
|
||
# 验证所有步骤都已添加
|
||
assert len(prcxi_9320_handler._unilabos_backend.steps_todo_list) == 4
|
||
functions = [step["Function"] for step in prcxi_9320_handler._unilabos_backend.steps_todo_list]
|
||
assert functions == ["Load", "Imbibing", "Tapping", "UnLoad"]
|
||
|
||
|
||
class TestPRCXILayoutRecommendation:
|
||
"""测试 PRCXI 板位推荐功能"""
|
||
|
||
def test_9300_layout_creation(self, default_layout_9300):
|
||
"""测试 PRCXI 9300 布局创建"""
|
||
layout_info = default_layout_9300.get_layout()
|
||
assert layout_info["rows"] == 2
|
||
assert layout_info["columns"] == 3
|
||
assert len(layout_info["layout"]) == 6
|
||
assert layout_info["trash_slot"] == 6
|
||
assert "waste_liquid_slot" not in layout_info
|
||
|
||
def test_9320_layout_creation(self, default_layout_9320):
|
||
"""测试 PRCXI 9320 布局创建"""
|
||
layout_info = default_layout_9320.get_layout()
|
||
assert layout_info["rows"] == 4
|
||
assert layout_info["columns"] == 4
|
||
assert len(layout_info["layout"]) == 16
|
||
assert layout_info["trash_slot"] == 16
|
||
assert layout_info["waste_liquid_slot"] == 12
|
||
|
||
def test_layout_recommendation_9320(self, default_layout_9320, prcxi_materials):
|
||
"""测试 PRCXI 9320 板位推荐功能"""
|
||
# 添加物料信息
|
||
default_layout_9320.add_lab_resource(prcxi_materials)
|
||
|
||
# 推荐布局
|
||
needs = [
|
||
("reagent_1", "96 细胞培养皿", 3),
|
||
("reagent_2", "12道储液槽", 1),
|
||
("reagent_3", "200μL Tip头", 7),
|
||
("reagent_4", "10μL加长 Tip头", 1),
|
||
]
|
||
|
||
matrix_layout, layout_list = default_layout_9320.recommend_layout(needs)
|
||
|
||
# 验证返回结果
|
||
assert "MatrixId" in matrix_layout
|
||
assert "MatrixName" in matrix_layout
|
||
assert "MatrixCount" in matrix_layout
|
||
assert "WorkTablets" in matrix_layout
|
||
assert len(layout_list) == 12 # 3+1+7+1 = 12个位置
|
||
|
||
# 验证推荐的位置不包含预留位置
|
||
reserved_positions = {12, 16}
|
||
recommended_positions = [item["positions"] for item in layout_list]
|
||
for pos in recommended_positions:
|
||
assert pos not in reserved_positions
|
||
|
||
def test_layout_recommendation_insufficient_space(self, default_layout_9320, prcxi_materials):
|
||
"""测试板位推荐空间不足的情况"""
|
||
# 添加物料信息
|
||
default_layout_9320.add_lab_resource(prcxi_materials)
|
||
|
||
# 尝试推荐超过可用空间的布局
|
||
needs = [
|
||
("reagent_1", "96 细胞培养皿", 15), # 需要15个位置，但只有14个可用
|
||
]
|
||
|
||
with pytest.raises(ValueError, match="需要 .* 个位置，但只有 .* 个可用位置"):
|
||
default_layout_9320.recommend_layout(needs)
|
||
|
||
def test_layout_recommendation_material_not_found(self, default_layout_9320, prcxi_materials):
|
||
"""测试板位推荐物料不存在的情况"""
|
||
# 添加物料信息
|
||
default_layout_9320.add_lab_resource(prcxi_materials)
|
||
|
||
# 尝试推荐不存在的物料
|
||
needs = [
|
||
("reagent_1", "不存在的物料", 1),
|
||
]
|
||
|
||
with pytest.raises(ValueError, match="Material .* not found in lab resources"):
|
||
default_layout_9320.recommend_layout(needs)
|
||
|
||
|
||
class TestPRCXIBackendOperations:
|
||
"""测试 PRCXI 后端操作功能"""
|
||
|
||
def test_backend_initialization(self, prcxi_9300_handler):
|
||
"""测试后端初始化"""
|
||
backend = prcxi_9300_handler._unilabos_backend
|
||
assert isinstance(backend, PRCXI9300Backend)
|
||
assert backend._num_channels == 8
|
||
assert backend.debug is True
|
||
|
||
def test_protocol_creation(self, prcxi_9300_handler):
|
||
"""测试协议创建"""
|
||
backend = prcxi_9300_handler._unilabos_backend
|
||
backend.create_protocol("Test Protocol")
|
||
assert backend.protocol_name == "Test Protocol"
|
||
assert len(backend.steps_todo_list) == 0
|
||
|
||
def test_channel_validation(self):
|
||
"""测试通道验证"""
|
||
# 测试正确的8通道配置
|
||
valid_channels = [0, 1, 2, 3, 4, 5, 6, 7]
|
||
result = PRCXI9300Backend.check_channels(valid_channels)
|
||
assert result == valid_channels
|
||
|
||
# 测试错误的通道配置
|
||
invalid_channels = [0, 1, 2, 3]
|
||
result = PRCXI9300Backend.check_channels(invalid_channels)
|
||
assert result == [0, 1, 2, 3, 4, 5, 6, 7]
|
||
|
||
def test_matrix_info_creation(self, prcxi_9300_handler):
|
||
"""测试矩阵信息创建"""
|
||
backend = prcxi_9300_handler._unilabos_backend
|
||
backend.create_protocol("Test Protocol")
|
||
|
||
# 模拟运行协议时的矩阵信息创建
|
||
run_time = 1234567890
|
||
matrix_info = MatrixInfo(
|
||
MatrixId=f"{int(run_time)}",
|
||
MatrixName=f"protocol_{run_time}",
|
||
MatrixCount=len(backend.tablets_info),
|
||
WorkTablets=backend.tablets_info,
|
||
)
|
||
|
||
assert matrix_info["MatrixId"] == str(int(run_time))
|
||
assert matrix_info["MatrixName"] == f"protocol_{run_time}"
|
||
assert "WorkTablets" in matrix_info
|
||
|
||
|
||
class TestPRCXIContainerOperations:
|
||
"""测试 PRCXI 容器操作功能"""
|
||
|
||
def test_container_serialization(self, tip_rack_300ul):
|
||
"""测试容器序列化"""
|
||
serialized = tip_rack_300ul.serialize_state()
|
||
assert "Material" in serialized
|
||
assert serialized["Material"]["Name"] == "300μL Tip头"
|
||
|
||
def test_container_deserialization(self, tip_rack_300ul):
|
||
"""测试容器反序列化"""
|
||
# 序列化
|
||
serialized = tip_rack_300ul.serialize_state()
|
||
|
||
# 创建新容器并反序列化
|
||
new_tip_rack = PRCXI9300Container(
|
||
name="new_tip_rack",
|
||
size_x=50,
|
||
size_y=50,
|
||
size_z=10,
|
||
category="tip_rack",
|
||
ordering=collections.OrderedDict()
|
||
)
|
||
new_tip_rack.load_state(serialized)
|
||
|
||
assert new_tip_rack._unilabos_state["Material"]["Name"] == "300μL Tip头"
|
||
|
||
def test_trash_container_creation(self, prcxi_materials):
|
||
"""测试垃圾桶容器创建"""
|
||
trash = PRCXI9300Trash(name="trash", size_x=50, size_y=50, size_z=10, category="trash")
|
||
trash.load_state({
|
||
"Material": {
|
||
"uuid": prcxi_materials["废弃槽"]["uuid"]
|
||
}
|
||
})
|
||
|
||
assert trash.name == "trash"
|
||
assert trash._unilabos_state["Material"]["uuid"] == prcxi_materials["废弃槽"]["uuid"]
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 运行测试
|
||
pytest.main([__file__, "-v"]) |