From 2027d898048457e7654bcd4685add6f5e09414f8 Mon Sep 17 00:00:00 2001
From: Jusong Yu <jusong.yeu@gmail.com>
Date: Wed, 14 Dec 2022 11:47:29 +0100
Subject: [PATCH] Update test_schema.py

---
 tests/test_schema.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/test_schema.py b/tests/test_schema.py
index 3d5d1ce..5f5018a 100644
--- a/tests/test_schema.py
+++ b/tests/test_schema.py
@@ -20,6 +20,7 @@ def bands_schema():
 def test_valid_bands_default(bands_schema):
     data = {
         "fermi_level": -7.0,
+        "color-partial bands (only one field is needed for this case.)": "blue",
         "path": [["GAMMA", "Y"], ["Y", "C_0"], ["SIGMA_0", "GAMMA"]],
         "paths": [
             {
@@ -31,6 +32,11 @@ def test_valid_bands_default(bands_schema):
                     [1.0, 1.1, 1.2],
                     [2.0, 2.1, 2.2],
                 ],
+                "highlight (thickness of the bands) (optional)": [
+                    [0.0, 0.1, 0.2],
+                    [1.0, 1.1, 1.2],
+                    [2.0, 2.1, 2.2],
+                ]
                 "x": [0.0, 1.0, 2.0],
             },
             {