Preserve empty static display names
This commit is contained in:
@@ -4939,6 +4939,33 @@ NODE_DISPLAY_NAME_MAPPINGS = {
|
|||||||
self.assertEqual({}, result["nodes"])
|
self.assertEqual({}, result["nodes"])
|
||||||
self.assertEqual("no_static_nodes", result["pack"]["status"])
|
self.assertEqual("no_static_nodes", result["pack"]["status"])
|
||||||
|
|
||||||
|
def test_empty_display_mapping_value_is_preserved(self):
|
||||||
|
source = '''
|
||||||
|
class EmptyDisplayNode:
|
||||||
|
RETURN_TYPES = ("IMAGE",)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def INPUT_TYPES(cls):
|
||||||
|
return {
|
||||||
|
"required": {
|
||||||
|
"image": ("IMAGE",),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
NODE_CLASS_MAPPINGS = {
|
||||||
|
"EmptyDisplayNode": EmptyDisplayNode,
|
||||||
|
}
|
||||||
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||||
|
"EmptyDisplayNode": "",
|
||||||
|
}
|
||||||
|
'''
|
||||||
|
result = self._extract_source(source, "empty-display-pack")
|
||||||
|
|
||||||
|
self.assertIn("EmptyDisplayNode", result["nodes"])
|
||||||
|
self.assertEqual("", result["nodes"]["EmptyDisplayNode"]["display"])
|
||||||
|
self.assertEqual("ok", result["pack"]["status"])
|
||||||
|
|
||||||
def test_non_string_display_mapping_value_skips_node(self):
|
def test_non_string_display_mapping_value_skips_node(self):
|
||||||
source = '''
|
source = '''
|
||||||
class NonStringDisplayValueNode:
|
class NonStringDisplayValueNode:
|
||||||
|
|||||||
@@ -2518,7 +2518,7 @@ def _signature_from_class(node_type, cls, display, pack_meta, class_env, input_e
|
|||||||
|
|
||||||
return {
|
return {
|
||||||
"type": node_type,
|
"type": node_type,
|
||||||
"display": display or node_type,
|
"display": display if display is not None else node_type,
|
||||||
"pack": pack_meta["id"],
|
"pack": pack_meta["id"],
|
||||||
"repository": pack_meta.get("repository", ""),
|
"repository": pack_meta.get("repository", ""),
|
||||||
"inputs": inputs,
|
"inputs": inputs,
|
||||||
|
|||||||
Reference in New Issue
Block a user