def test_remove_K3_D2(self, K=3, D=2):
    A = ParamBag(K=K, D=D)
    s = 123
    N = np.random.rand(K)
    x = np.random.rand(K,D)
    xxT = np.random.randn(K,D,D)
    A.setField('s', s)
    A.setField('N', N, dims='K')
    A.setField('x', x, dims=('K','D'))
    A.setField('xxT', xxT, dims=('K','D','D'))
    Abig = A.copy()
    # First remove a few fields
    for k in range(K-1):
      A.removeComp(0)
      assert A.K == K - k - 1
      assert A.s == s
      assert np.allclose(A.getComp(0).x, x[k+1])
      assert np.allclose(A.getComp(0).xxT, xxT[k+1])

  ######################################################### Verify get
dataset/ETHPy150Open daeilkim/refinery/refinery/bnpy/bnpy-dev/tests/suffstats/TestParamBag.py/TestParamBag.test_remove_K3_D2/original