fix tests

Signed-off-by: shanjiaz <[email protected]>

Author: shanjiaz

tests/test_compressors/quantized_compressors/test_pack_quant.py

@@ -88,9 +88,8 @@ def test_quant_format(shape):
         dense_state_dict, names_to_scheme=quantized_modules_to_scheme
     )
 
-    # compressed state_dict adds one entry for shape
-    # but removes the zero points since we are symmetric
-    assert len(dense_state_dict) == len(compressed_state_dict)
+    # compressed state_dict adds one entry for shape and keeps zero_point
+    assert len(dense_state_dict) + 1 == len(compressed_state_dict)
 
     # check compressed and packed
     assert compressed_state_dict["dummy.weight_packed"].dtype == torch.int32

tests/test_compressors/quantized_compressors/test_packed_asym_decompression.py

@@ -114,6 +114,7 @@ def test_end_to_end_asymmetric_quantization(
     # Verify compression created zero-point parameters
     assert hasattr(model.layer1, "weight_zero_point")
     assert hasattr(model.layer2, "weight_zero_point")
+    # For asymmetric GROUP/CHANNEL quantization, zero_point should be packed to int32
     assert model.layer1.weight_zero_point.dtype == torch.int32
     assert model.layer2.weight_zero_point.dtype == torch.int32