Browse Source

Add unit test for style (#615)

Xiaojing Chen 2 years ago
parent
commit
55ac6e4973
1 changed files with 39 additions and 0 deletions
  1. 39 0
      tests/test_style.py

+ 39 - 0
tests/test_style.py

@@ -0,0 +1,39 @@
+import pytest
+
+from pynecone import style
+from pynecone.var import Var
+
+test_style = [
+    ({"a": 1}, {"a": 1}),
+    ({"a": Var.create("abc")}, {"a": "abc"}),
+    ({"test_case": 1}, {"testCase": 1}),
+    ({"test_case": {"a": 1}}, {"testCase": {"a": 1}}),
+]
+
+
+@pytest.mark.parametrize(
+    "style_dict,expected",
+    test_style,
+)
+def test_convert(style_dict, expected):
+    """Test Format a style dictionary.
+
+    Args:
+        style_dict: The style to check.
+        expected: The expected formatted style.
+    """
+    assert style.convert(style_dict) == expected
+
+
+@pytest.mark.parametrize(
+    "style_dict,expected",
+    test_style,
+)
+def test_create_style(style_dict, expected):
+    """Test style dictionary.
+
+    Args:
+        style_dict: The style to check.
+        expected: The expected formatted style.
+    """
+    assert style.Style(style_dict) == expected