mirror of
https://github.com/dptech-corp/Uni-Lab-OS.git
synced 2025-12-17 13:01:12 +00:00
77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
import pytest
|
|
import json
|
|
import os
|
|
|
|
from pylabrobot.resources import Resource as ResourcePLR
|
|
from unilabos.resources.graphio import resource_bioyond_to_plr
|
|
from unilabos.registry.registry import lab_registry
|
|
|
|
from unilabos.resources.bioyond.decks import BIOYOND_PolymerReactionStation_Deck
|
|
|
|
lab_registry.setup()
|
|
|
|
|
|
type_mapping = {
|
|
"烧杯": "BIOYOND_PolymerStation_1FlaskCarrier",
|
|
"试剂瓶": "BIOYOND_PolymerStation_1BottleCarrier",
|
|
"样品板": "BIOYOND_PolymerStation_6VialCarrier",
|
|
"分装板": "BIOYOND_PolymerStation_6VialCarrier",
|
|
"样品瓶": "BIOYOND_PolymerStation_Solid_Stock",
|
|
"90%分装小瓶": "BIOYOND_PolymerStation_Solid_Vial",
|
|
"10%分装小瓶": "BIOYOND_PolymerStation_Liquid_Vial",
|
|
}
|
|
|
|
type_uuid_mapping = {
|
|
"烧杯": "BIOYOND_PolymerStation_1FlaskCarrier",
|
|
"试剂瓶": "BIOYOND_PolymerStation_1BottleCarrier",
|
|
"样品板": "BIOYOND_PolymerStation_6VialCarrier",
|
|
"分装板": "3a14196e-5dfe-6e21-0c79-fe2036d052c4",
|
|
"样品瓶": "3a14196a-cf7d-8aea-48d8-b9662c7dba94",
|
|
"90%分装小瓶": "3a14196c-cdcf-088d-dc7d-5cf38f0ad9ea",
|
|
"10%分装小瓶": "3a14196c-76be-2279-4e22-7310d69aed68",
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def bioyond_materials_reaction() -> list[dict]:
|
|
print("加载 BioYond 物料数据...")
|
|
print(os.getcwd())
|
|
with open("bioyond_materials_reaction.json", "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
print(f"加载了 {len(data)} 条物料数据")
|
|
return data
|
|
|
|
|
|
@pytest.fixture
|
|
def bioyond_materials_liquidhandling_1() -> list[dict]:
|
|
print("加载 BioYond 物料数据...")
|
|
print(os.getcwd())
|
|
with open("bioyond_materials_liquidhandling_1.json", "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
print(f"加载了 {len(data)} 条物料数据")
|
|
return data
|
|
|
|
|
|
@pytest.fixture
|
|
def bioyond_materials_liquidhandling_2() -> list[dict]:
|
|
print("加载 BioYond 物料数据...")
|
|
print(os.getcwd())
|
|
with open("bioyond_materials_liquidhandling_2.json", "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
print(f"加载了 {len(data)} 条物料数据")
|
|
return data
|
|
|
|
|
|
@pytest.mark.parametrize("materials_fixture", [
|
|
"bioyond_materials_reaction",
|
|
"bioyond_materials_liquidhandling_1",
|
|
])
|
|
def test_bioyond_to_plr(materials_fixture, request) -> list[dict]:
|
|
materials = request.getfixturevalue(materials_fixture)
|
|
deck = BIOYOND_PolymerReactionStation_Deck("test_deck")
|
|
output = resource_bioyond_to_plr(materials, type_mapping=type_mapping, deck=deck)
|
|
print(deck.summary())
|
|
print([resource.serialize() for resource in output])
|
|
print([resource.serialize_all_state() for resource in output])
|
|
json.dump(deck.serialize(), open("test.json", "w", encoding="utf-8"), indent=4)
|