Add automated unit tests

tests/llmcompressor/recipe/test_recipe.py

@@ -3,8 +3,10 @@
 import pytest
 import yaml
 
+from llmcompressor.modifiers import Modifier
 from llmcompressor.modifiers.obcq.base import SparseGPTModifier
 from llmcompressor.recipe import Recipe
+from llmcompressor.recipe.recipe import create_recipe_string_from_modifiers
 from tests.llmcompressor.helpers import valid_recipe_strings
 
 
@@ -96,3 +98,45 @@ def test_recipe_can_be_created_from_modifier_instances():
     ):
         assert isinstance(actual_modifier, type(expected_modifier))
         assert actual_modifier.dict() == expected_modifier.dict()
+
+
+class A_FirstDummyModifier(Modifier):
+    def model_dump(self):
+        return {}
+
+
+class B_SecondDummyModifier(Modifier):
+    def model_dump(self):
+        return {}
+
+
+def test_create_recipe_string_from_modifiers_with_default_group_name():
+    modifiers = [B_SecondDummyModifier(), A_FirstDummyModifier()]
+    expected_recipe_str = (
+        "DEFAULT_stage:\n"
+        "  DEFAULT_modifiers:\n"
+        "    B_SecondDummyModifier: {}\n"
+        "    A_FirstDummyModifier: {}\n"
+    )
+    actual_recipe_str = create_recipe_string_from_modifiers(modifiers)
+    assert actual_recipe_str == expected_recipe_str
+
+
+def test_create_recipe_string_from_modifiers_with_custom_group_name():
+    modifiers = [B_SecondDummyModifier(), A_FirstDummyModifier()]
+    group_name = "custom"
+    expected_recipe_str = (
+        "custom_stage:\n"
+        "  DEFAULT_modifiers:\n"
+        "    B_SecondDummyModifier: {}\n"
+        "    A_FirstDummyModifier: {}\n"
+    )
+    actual_recipe_str = create_recipe_string_from_modifiers(modifiers, group_name)
+    assert actual_recipe_str == expected_recipe_str
+
+
+def test_create_recipe_string_from_modifiers_with_empty_modifiers():
+    modifiers = []
+    expected_recipe_str = "DEFAULT_stage:\n" "  DEFAULT_modifiers: {}\n"
+    actual_recipe_str = create_recipe_string_from_modifiers(modifiers)
+    assert actual_recipe_str == expected_recipe_str