()
| 1200 | |
| 1201 | |
| 1202 | def test_cast_array_to_features_sequence_classlabel(): |
| 1203 | arr = pa.array([[], [1], [0, 1]], pa.list_(pa.int64())) |
| 1204 | assert cast_array_to_feature(arr, List(ClassLabel(names=["foo", "bar"]))).type == pa.list_(pa.int64()) |
| 1205 | |
| 1206 | arr = pa.array([[], ["bar"], ["foo", "bar"]], pa.list_(pa.string())) |
| 1207 | assert cast_array_to_feature(arr, List(ClassLabel(names=["foo", "bar"]))).type == pa.list_(pa.int64()) |
| 1208 | |
| 1209 | # Test empty arrays |
| 1210 | arr = pa.array([[], []], pa.list_(pa.int64())) |
| 1211 | assert cast_array_to_feature(arr, List(ClassLabel(names=["foo", "bar"]))).type == pa.list_(pa.int64()) |
| 1212 | |
| 1213 | arr = pa.array([[], []], pa.list_(pa.string())) |
| 1214 | assert cast_array_to_feature(arr, List(ClassLabel(names=["foo", "bar"]))).type == pa.list_(pa.int64()) |
| 1215 | |
| 1216 | # Test invalid class labels |
| 1217 | arr = pa.array([[2]], pa.list_(pa.int64())) |
| 1218 | with pytest.raises(ValueError): |
| 1219 | assert cast_array_to_feature(arr, List(ClassLabel(names=["foo", "bar"]))) |
| 1220 | |
| 1221 | arr = pa.array([["baz"]], pa.list_(pa.string())) |
| 1222 | with pytest.raises(ValueError): |
| 1223 | assert cast_array_to_feature(arr, List(ClassLabel(names=["foo", "bar"]))) |
| 1224 | |
| 1225 | |
| 1226 | @pytest.mark.parametrize( |
nothing calls this directly
no test coverage detected