def METHOD_NAME(model, ref_trainable_weights, ref_non_trainable_weights):
    for layer in model.layers:
        if isinstance(layer, NNCFWrapper):
            actual_trainable_weights = [weight.name for weight in layer.trainable_weights]
            actual_non_trainable_weights = [weight.name for weight in layer.non_trainable_weights]
            assert actual_trainable_weights == ref_trainable_weights
            assert actual_non_trainable_weights == ref_non_trainable_weights