22 lines
512 B
Python
22 lines
512 B
Python
from atst.models.task_order import TaskOrder
|
|
|
|
from tests.factories import TaskOrderFactory
|
|
|
|
|
|
def test_as_dictionary():
|
|
data = TaskOrderFactory.dictionary()
|
|
real_task_order = TaskOrderFactory.create(**data)
|
|
assert real_task_order.to_dictionary() == data
|
|
|
|
|
|
def test_budget():
|
|
task_order = TaskOrderFactory.create(
|
|
clin_0001=500,
|
|
clin_0003=200,
|
|
clin_1001=None,
|
|
clin_1003=None,
|
|
clin_2001=None,
|
|
clin_2003=None,
|
|
)
|
|
assert task_order.budget == 700
|