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))