[Fix] remove tests/quantization (vllm-project#99)

* initial commit * woops --------- Co-authored-by: [email protected] <[email protected]>
markmc · Jul 1, 2024 · cff8a10 · cff8a10
1 parent 3db8471
commit cff8a10
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 46 deletions.
diff --git a/tests/quantization/__init__.py b/tests/quantization/__init__.py
diff --git a/tests/quantization/test_quant_config.py b/tests/quantization/test_quant_config.py
diff --git a/tests/test_quantization/test_quant_config.py b/tests/test_quantization/test_quant_config.py
@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
 import pytest
 from compressed_tensors.quantization import (
     DEFAULT_QUANTIZATION_FORMAT,
@@ -60,3 +59,19 @@ def test_full_config():
 def test_need_config_groups():
     with pytest.raises(ValidationError):
         _ = QuantizationScheme()
+
+
+@pytest.mark.parametrize(
+    "scheme_name",
+    [
+        "W8A8",
+        "W4A16",
+    ],
+)
+def test_load_scheme_from_preset(scheme_name: str):
+    targets = ["Linear"]
+    config = QuantizationConfig(config_groups={scheme_name: targets})
+
+    assert scheme_name in config.config_groups
+    assert isinstance(config.config_groups[scheme_name], QuantizationScheme)
+    assert config.config_groups[scheme_name].targets == targets