4392d120d845adf25d96d353df1513aa8e1df0ad,tests/test_loss_weighting.py,,test_class_weight_wrong_classes,#,152

Before Change

    (x_train, y_train), (x_test, y_test), (sample_weight, class_weight, test_ids) = _get_test_data()

    del class_weight[1]
    try:
        model.fit(x_train, y_train, epochs=0, verbose=0, class_weight=class_weight)
        assert False
    except ValueError:
        pass  // expected behavior is to raise a ValueError with a suitable message


if __name__ == "__main__":
    pytest.main([__file__])

After Change

    (x_train, y_train), (x_test, y_test), (sample_weight, class_weight, test_ids) = _get_test_data()

    del class_weight[1]
    with pytest.raises(ValueError):
        model.fit(x_train, y_train,
                  epochs=0, verbose=0, class_weight=class_weight)


if __name__ == "__main__":
    pytest.main([__file__])

Instances