Skip to content

Commit 49f63a2

Browse files
committed
update test to updated problem class organisation
1 parent 6bf94f8 commit 49f63a2

1 file changed

Lines changed: 8 additions & 13 deletions

File tree

tests/make_realistic/problems/test_scenarios.py

Lines changed: 8 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
from dataclasses import is_dataclass
21
from datetime import timedelta
32

43
from virtualship.instruments.types import InstrumentType
@@ -12,28 +11,24 @@
1211

1312
def _assert_general_problem_class(cls):
1413
assert isinstance(cls, GeneralProblem)
15-
instance = cls()
16-
assert is_dataclass(instance)
1714

1815
# required attributes and types
19-
assert isinstance(instance.message, str)
20-
assert instance.message.strip(), "message should not be empty"
16+
assert isinstance(cls.message, str)
17+
assert cls.message.strip(), "message should not be empty"
2118

22-
assert isinstance(instance.delay_duration, timedelta)
23-
assert isinstance(instance.pre_departure, bool)
19+
assert isinstance(cls.delay_duration, timedelta)
20+
assert isinstance(cls.pre_departure, bool)
2421

2522

2623
def _assert_instrument_problem_class(cls):
2724
assert isinstance(cls, InstrumentProblem)
28-
instance = cls()
29-
assert is_dataclass(instance)
3025

3126
# required attributes and types
32-
assert isinstance(instance.message, str)
33-
assert instance.message.strip(), "message should not be empty"
27+
assert isinstance(cls.message, str)
28+
assert cls.message.strip(), "message should not be empty"
3429

35-
assert isinstance(instance.delay_duration, timedelta)
36-
assert isinstance(instance.instrument_type, InstrumentType)
30+
assert isinstance(cls.delay_duration, timedelta)
31+
assert isinstance(cls.instrument_type, InstrumentType)
3732

3833

3934
def test_general_problems():

0 commit comments

Comments
 (0)