def test_json_stage4_should_be_model_stage4(self):
stage = Stage(level=4, name="守門人的意思", stamina=40, battles=10, reward=0, experience=5000)
stage_json = [{"level": 4, "name": "守門人的意思", "attribute": ["水", "火", "暗"], \
"stamina": 40, "battles": 10, "reward": 0, "experience": 5000}]
stages = Stage.transform_json(stage_json)
self.assertEqual(stage, stages[0])
def test_json_with_wrong_format_should_not_create_model(self):
stage_json = [{"level": 4}]
stages = Stage.transform_json(stage_json)
self.assertEqual(0, len(stages))
