Add unit tests for param_mapping.py#3711

bvandermoon wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

bvandermoon-param-mapping-testsAI-Hypercomputer/maxtext:bvandermoon-param-mapping-testsCopy head branch name to clipboard