Adds support for adding examples inside comments

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/test_modifier.py b/tests/test_modifier.py
index 1e109d8..8acb594 100644
--- a/tests/test_modifier.py
+++ b/tests/test_modifier.py
@@ -13,3 +13,10 @@ class ModiferTestCase(unittest.TestCase):
 
     def test_handles_multiple_classes(self):
         self.assertTrue('callout extreme' in self.modifier.class_name)
+
+    def test_add_example(self):
+        example = '<i class="icon$modifier_class"></i>'
+        expected = '<i class="icon callout extreme pseudo-class-hover"></i>'
+
+        self.modifier.add_example(example)
+        self.assertEqual(self.modifier.example, expected)