Commit 8011a52f authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

some tests

parent 1d237193
import pytest from csst_dag.dag import CSST_DAG_LIST
from csst_dag import gen_level1_dag_message
dag = CSST_DAG_LIST[0]
print(dag)
def test_gen_level1_dag_message_msc_sls(): dag.pprint()
msg = gen_level1_dag_message(obsid="11009101682009", chipid="01", return_dict=True) dag.match(obsid="11009101682009", chipid="01")
assert msg["dag_id"] == "csst-msc-l1-sls" dag.match(obsid="10109101682009", chipid="12")
def test_gen_level1_dag_message_msc_mbi(): assert CSST_DAG_LIST.match(obsid="11009101682009", chipid="01") == ["csst-msc-l1-sls"]
msg = gen_level1_dag_message(obsid="11009101682009", chipid="09", return_dict=True) assert CSST_DAG_LIST.match(obsid="11009101682009", chipid="12") == [
assert msg["dag_id"] == "csst-msc-l1-mbi" "csst-msc-l1-mbi",
"csst-msc-l1-ast",
]
assert CSST_DAG_LIST.match(obsid="12009101682009", chipid="09") == ["csst-msc-l1-qc0"]
def test_gen_level1_dag_message_msc_invalid_chipid(): assert CSST_DAG_LIST.match(obsid="12009101682009", chipid="00") == []
with pytest.raises(ValueError):
msg = gen_level1_dag_message(
obsid="11009101682009", chipid="00", return_dict=True
)
assert CSST_DAG_LIST.match(a=1) == []
def test_gen_level1_dag_message_msc_none():
msg = gen_level1_dag_message(obsid="12009101682009", chipid="00", return_dict=True) dags = CSST_DAG_LIST.match_dag(obsid="11009101682009", chipid="01")
assert msg is None
dag = dags[0]
dag.gen_message(obsid="11009101682009", chipid="01")
import pytest
from csst_dag import gen_level1_dag_message
def test_gen_level1_dag_message_msc_sls():
msg = gen_level1_dag_message(obsid="11009101682009", chipid="01", return_dict=True)
assert msg["dag_id"] == "csst-msc-l1-sls"
def test_gen_level1_dag_message_msc_mbi():
msg = gen_level1_dag_message(obsid="11009101682009", chipid="09", return_dict=True)
assert msg["dag_id"] == "csst-msc-l1-mbi"
def test_gen_level1_dag_message_msc_invalid_chipid():
with pytest.raises(ValueError):
msg = gen_level1_dag_message(
obsid="11009101682009", chipid="00", return_dict=True
)
def test_gen_level1_dag_message_msc_none():
msg = gen_level1_dag_message(obsid="12009101682009", chipid="00", return_dict=True)
assert msg is None
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment