Update tests/unit/torch/export/test_nvfp4_utils.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: ynankani-nv <ynankani@nvidia.com>

Author: ynankani

tests/unit/torch/export/test_nvfp4_utils.py

@@ -89,6 +89,13 @@ def test_row_col_padding(self):
         assert s.shape[0] % 16 == 0
         assert s.shape[1] % 16 == 0
 
+    def test_row_col_padding_scale_only(self):
+        sd = _make_nvfp4_state_dict(rows=32, cols=48)
+        result = pad_nvfp4_weights(sd, "row_col")
+
+        assert result["layer0.weight"].shape == (32, 48)
+        assert result["layer0.weight_scale"].shape == (32, 16)
+
     def test_already_aligned(self):
         sd = _make_nvfp4_state_dict(rows=32, cols=64)
         orig_w_shape = sd["layer0.weight"].shape