code: def test_load_aids():
      x, y = sdata.load_aids(endpoint="aids")
      assert x.shape == (1151, 11)
      assert y.shape == (1151,)
      assert_structured_array_dtype(y, "censor", "time", 96)
      assert "censor_d" not in x.columns
      assert "time_d" not in x.columns
  
      x, y = sdata.load_aids(endpoint="death")
      assert x.shape == (1151, 11)
      assert y.shape == (1151,)
      assert_structured_array_dtype(y, "censor_d", "time_d", 26)
      assert "censor" not in x.columns
      assert "time" not in x.columns
  
      with pytest.raises(
          ValueError, match="endpoint must be 'aids' or 'death'"
      ):
          sdata.load_aids(endpoint="foobar")
  
imports: from contextlib import ExitStack
  from contextlib import nullcontext
  from io import StringIO
  import os
  import tempfile
  import numpy
  from numpy.testing import assert_array_almost_equal, assert_array_equal
  import pandas
  import pandas.testing
  import pytest
  import sksurv.datasets
  from sksurv.io import writearff
  from sksurv.testing import FixtureParameterFactory
label_desc: Use image node_id 4 for calling the TestLoadDatasets obj's underlying member method code with example usage: obj.test_load_aids() without return types