Small refinements for JSON model. (#5112)

* Naming consistency. * Remove duplicated test.
2019-12-11 19:49:01 +08:00
parent 208ab3b1ff
commit ad4a1c732c
7 changed files with 19 additions and 30 deletions
--- a/tests/python/test_basic_models.py
+++ b/tests/python/test_basic_models.py
@@ -213,12 +213,12 @@ class TestModels(unittest.TestCase):

        with open('./model.json', 'r') as fd:
            j_model = json.load(fd)
-        assert isinstance(j_model['Learner'], dict)
+        assert isinstance(j_model['learner'], dict)

        bst = xgb.Booster(model_file='./model.json')

        with open('./model.json', 'r') as fd:
            j_model = json.load(fd)
-        assert isinstance(j_model['Learner'], dict)
+        assert isinstance(j_model['learner'], dict)

        os.remove('model.json')