Method test_normalize_gpu_ids

tests/test_config.py:70–82

(self)

            self.assertEqual(parse_gpu_ids(ids), target)

    def test_normalize_gpu_ids(self):
        for ids, target in [
            (None, None),
            ("0,1,2", ["0", "1", "2"]),
            ("MIG-f1e$%^", ["MIG-f1e"]),
            ([1, 2, 3], [1, 2, 3]),
            (
                [0, "MIG-f1e$%^", "1, 2", "GPU-abcd!"],
                [0, "MIG-f1e", "1", "2", "GPU-abcd"],
            ),
            ([0, {"invalid": "entry"}, "GPU-123"], [0, "GPU-123"]),
        ]:
            self.assertEqual(normalize_gpu_ids(ids), target)

    @mock.patch.dict(
        os.environ,