| 5345 | |
| 5346 | |
| 5347 | def test_get_metadata_notrain(): |
| 5348 | model = CatBoost() |
| 5349 | with pytest.raises(CatBoostError): |
| 5350 | model.get_metadata()[1] = '1' |
| 5351 | pytest.fail('Only string keys should be allowed') |
| 5352 | with pytest.raises(CatBoostError): |
| 5353 | model.get_metadata()['1'] = 1 |
| 5354 | pytest.fail('Only string values should be allowed') |
| 5355 | model.get_metadata()['1'] = '1' |
| 5356 | assert model.get_metadata().get('1', 'EMPTY') == '1' |
| 5357 | assert model.get_metadata().get('2', 'EMPTY') == 'EMPTY' |
| 5358 | for i in range(100): |
| 5359 | model.get_metadata()[str(i)] = str(i) |
| 5360 | del model.get_metadata()['98'] |
| 5361 | with pytest.raises(KeyError): |
| 5362 | i = model.get_metadata()['98'] |
| 5363 | for i in range(0, 98, 2): |
| 5364 | assert str(i) in model.get_metadata() |
| 5365 | del model.get_metadata()[str(i)] |
| 5366 | for i in range(0, 98, 2): |
| 5367 | assert str(i) not in model.get_metadata() |
| 5368 | assert str(i + 1) in model.get_metadata() |
| 5369 | |
| 5370 | |
| 5371 | def test_metadata(): |