Remove duplicate IO Struct test (#6180)
Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
This commit is contained in:
@@ -124,24 +124,6 @@ class TestGenerateReqInputNormalization(CustomTestCase):
|
|||||||
# Ensure that text items are properly duplicated too
|
# Ensure that text items are properly duplicated too
|
||||||
self.assertEqual(req.text, expected_text)
|
self.assertEqual(req.text, expected_text)
|
||||||
|
|
||||||
def test_list_of_lists_with_none_values(self):
|
|
||||||
"""Test handling of list of lists with None values."""
|
|
||||||
req = copy.deepcopy(self.base_req)
|
|
||||||
req.image_data = [
|
|
||||||
[None], # None value
|
|
||||||
["image.jpg"], # Single image
|
|
||||||
]
|
|
||||||
|
|
||||||
req.normalize_batch_and_arguments()
|
|
||||||
|
|
||||||
# Structure should remain the same
|
|
||||||
self.assertEqual(len(req.image_data), 2)
|
|
||||||
self.assertEqual(len(req.image_data[0]), 1)
|
|
||||||
self.assertEqual(len(req.image_data[1]), 1)
|
|
||||||
|
|
||||||
# Check modalities
|
|
||||||
self.assertEqual(req.modalities, [None, "image"])
|
|
||||||
|
|
||||||
def test_specific_parallel_n_per_sample(self):
|
def test_specific_parallel_n_per_sample(self):
|
||||||
"""Test parallel expansion when different samples have different n values."""
|
"""Test parallel expansion when different samples have different n values."""
|
||||||
req = copy.deepcopy(self.base_req)
|
req = copy.deepcopy(self.base_req)
|
||||||
|
|||||||
Reference in New Issue
Block a user