Add AR, AG, AB test case and fix behavior

This commit is contained in:
Roy Nieterau 2025-05-19 12:04:01 +02:00
parent 526e5bfabb
commit 5917671521
2 changed files with 24 additions and 1 deletions

View file

@ -82,6 +82,29 @@ class GetReviewInfoByLayerName(unittest.TestCase):
}
}])
def test_ar_ag_ab_channels(self):
info = get_review_info_by_layer_name(["AR", "AG", "AB"])
self.assertEqual(info, [{
"name": "",
"review_channels": {
"R": "AR",
"G": "AG",
"B": "AB",
"A": None,
}
}])
info = get_review_info_by_layer_name(["AR", "AG", "AB", "A"])
self.assertEqual(info, [{
"name": "",
"review_channels": {
"R": "AR",
"G": "AG",
"B": "AB",
"A": "A",
}
}])
def test_unknown_channels(self):
info = get_review_info_by_layer_name(["hello", "world"])