CommencerCommencez gratuitement

Tester les schémas

L'équipe a corrigé les valeurs dans son DataFrame expected, mais Polars a inféré total_checkouts en 64 bits alors que le pipeline produit du 32 bits. Les deux schémas sont affichés pour que vous puissiez voir l'écart. Corrigez le type attendu, puis vérifiez que les schémas correspondent.

actual, expected et assert_schema_equal sont préchargés.

Cet exercice fait partie du cours

<cours>Mise à l'échelle et optimisation des pipelines de données avec Polars</cours>
Voir le cours

Instructions de l’exercice

  • Convertissez total_checkouts dans expected en pl.Int32.
  • Vérifiez que les schémas de actual et expected sont identiques.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Match the dtype the pipeline produces
expected = expected.with_columns(
    pl.col("total_checkouts").cast(pl.____)
)

# Schema-level assertion
____(actual.schema, expected.schema)
print("Schema assertion passed.")
print(actual.schema)
Modifier et exécuter le code