refactor: restore two-port concat backend schema
This commit is contained in:
@@ -31,7 +31,7 @@ class TestConcatByTargetNode(unittest.TestCase):
|
||||
self.assertLessEqual(float(image.max().item()), 1.0)
|
||||
self.assertTrue(torch.allclose(mask, torch.zeros_like(mask)))
|
||||
|
||||
def test_new_port_names_are_accepted(self):
|
||||
def test_legacy_port_names_are_accepted(self):
|
||||
image_a = self.make_image(width=2, height=2, value=0.2)
|
||||
image_b = self.make_image(width=1, height=2, value=0.8)
|
||||
|
||||
|
||||
@@ -41,10 +41,10 @@ class TestConcatByTargetRegistration(unittest.TestCase):
|
||||
self.assertEqual(required["background_value"][1]["default"], 0.0)
|
||||
self.assertEqual(required["multiple_of"][0], "INT")
|
||||
self.assertEqual(required["allow_batch_broadcast"][0], "BOOLEAN")
|
||||
self.assertEqual(optional["image/mask_A"][0], "IMAGE,MASK")
|
||||
self.assertEqual(optional["image/mask_B"][0], "IMAGE,MASK")
|
||||
self.assertNotIn("a", optional)
|
||||
self.assertNotIn("b", optional)
|
||||
self.assertEqual(str(optional["a"][0]), "*")
|
||||
self.assertEqual(str(optional["b"][0]), "*")
|
||||
self.assertNotIn("image/mask_A", optional)
|
||||
self.assertNotIn("image/mask_B", optional)
|
||||
self.assertNotIn("image_a", optional)
|
||||
self.assertNotIn("image_b", optional)
|
||||
self.assertNotIn("mask_a", optional)
|
||||
|
||||
Reference in New Issue
Block a user