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

some tests

parent 1d237193

tests/test_dag.py

+19 −16
Original line number Diff line number Diff line 
import pytest

from csst_dag import gen_level1_dag_message

from csst_dag.dag import CSST_DAG_LIST



dag = CSST_DAG_LIST[0]

print(dag)



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"

dag.pprint()

dag.match(obsid="11009101682009", chipid="01")

dag.match(obsid="10109101682009", chipid="12")





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"

assert CSST_DAG_LIST.match(obsid="11009101682009", chipid="01") == ["csst-msc-l1-sls"]

assert CSST_DAG_LIST.match(obsid="11009101682009", chipid="12") == [

    "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():

    with pytest.raises(ValueError):

        msg = gen_level1_dag_message(

            obsid="11009101682009", chipid="00", return_dict=True

        )

assert CSST_DAG_LIST.match(obsid="12009101682009", chipid="00") == []



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)

    assert msg is None
 


dags = CSST_DAG_LIST.match_dag(obsid="11009101682009", chipid="01")



dag = dags[0]

dag.gen_message(obsid="11009101682009", chipid="01")

tests/test_dag_message.py

0 → 100644
+24 −0
Original line number Diff line number Diff line 
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