@Test
public void testRemoveTreatment() {

    client
        .createExperiment("exp")
        .addTreatment("control")
        .apply();
    client.removeTreatment("exp", "control");
    assertEquals(0, client.getTreatments("exp").size());
}