def test_get_dense(): B_ref = np.array([ [0.0, 3.0], [2.0, 7.0], [5.0, 0.0] ]) assert np.allclose(get_dense(B_ref, 1, 2), np.array([[2.0, 3.0], [5.0, 7.0]]))