struct cs_dsp_mock_bin_builder *builder;
        struct wmfw_coeff_hdr *hdr;
 
+       KUNIT_ASSERT_LE(priv->test, format_version, 0xff);
+       KUNIT_ASSERT_LE(priv->test, fw_version, 0xffffff);
+
        builder = kunit_kzalloc(priv->test, sizeof(*builder), GFP_KERNEL);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(priv->test, builder);
        builder->test_priv = priv;
 
                 * Could be one 32-bit register or two 16-bit registers.
                 * A raw read will read the requested number of bytes.
                 */
-               regmap_raw_read(priv->dsp->regmap,
-                               xm + (offsetof(struct wmfw_adsp2_id_hdr, n_algs) / 2),
-                               &num_algs_be32, sizeof(num_algs_be32));
+               KUNIT_ASSERT_GE(priv->test, 0,
+                               regmap_raw_read(priv->dsp->regmap,
+                                               xm +
+                                               (offsetof(struct wmfw_adsp2_id_hdr, n_algs) / 2),
+                                               &num_algs_be32, sizeof(num_algs_be32)));
                num_algs = be32_to_cpu(num_algs_be32);
                bytes = sizeof(struct wmfw_adsp2_id_hdr) +
                        (num_algs * sizeof(struct wmfw_adsp2_alg_hdr)) +
                regcache_drop_region(priv->dsp->regmap, xm, xm + (bytes / 2) - 1);
                break;
        case WMFW_HALO:
-               regmap_read(priv->dsp->regmap,
-                           xm + offsetof(struct wmfw_halo_id_hdr, n_algs),
-                           &num_algs);
+               KUNIT_ASSERT_GE(priv->test, 0,
+                               regmap_read(priv->dsp->regmap,
+                                           xm + offsetof(struct wmfw_halo_id_hdr, n_algs),
+                                           &num_algs));
                bytes = sizeof(struct wmfw_halo_id_hdr) +
                        (num_algs * sizeof(struct wmfw_halo_alg_hdr)) +
                        4 /* terminator word */;
 
        size_t bytes_needed, name_len, description_len;
        int offset;
 
+       KUNIT_ASSERT_LE(builder->test_priv->test, alg_id, 0xffffff);
+
        /* Bytes needed for region header */
        bytes_needed = offsetof(struct wmfw_region, data);
 
 {
        struct cs_dsp_mock_wmfw_builder *builder;
 
+       KUNIT_ASSERT_LE(priv->test, format_version, 0xff);
+
        /* If format version isn't given use the default for the target core */
        if (format_version < 0) {
                switch (priv->dsp->type) {