49 lines
1.4 KiB
Java
49 lines
1.4 KiB
Java
package net.woodyfolsom.cs6601.p2;
|
|
|
|
import static org.junit.Assert.assertNotNull;
|
|
import static org.junit.Assert.assertThat;
|
|
import static org.hamcrest.core.IsEqual.equalTo;
|
|
import static org.hamcrest.core.Is.*;
|
|
|
|
import java.io.File;
|
|
|
|
import org.junit.Test;
|
|
|
|
public class SurveyReaderTest {
|
|
|
|
@Test
|
|
public void testReadLongSurveyDataset() {
|
|
Survey survey = SurveyReader.readSurvey(new File("data/long_survey.xml"));
|
|
assertNotNull(survey);
|
|
|
|
assertThat(survey.getDishCount(), equalTo(10));
|
|
assertThat(survey.getDinerCount(), equalTo(4));
|
|
|
|
assertThat(survey.isCategory(0,"vegetarian"), is(false));
|
|
assertThat(survey.isCategory(0,"allergic-nuts"), is(false));
|
|
|
|
assertThat(survey.isCategory(1,"vegetarian"), is(true));
|
|
assertThat(survey.isCategory(1,"allergic-nuts"), is(false));
|
|
|
|
assertThat(survey.isCategory(2,"vegan"), is(false));
|
|
assertThat(survey.isCategory(2,"allergic-nuts"), is(true));
|
|
|
|
assertThat(survey.isDiner("vegetarian"), is(true));
|
|
assertThat(survey.isDiner("allergic-nuts"), is(true));
|
|
assertThat(survey.isDiner("rastafarian"), is(false));
|
|
}
|
|
|
|
@Test
|
|
public void testReadMediumSurveyDataset() {
|
|
Survey survey = SurveyReader.readSurvey(new File("data/medium_survey.xml"));
|
|
assertNotNull(survey);
|
|
|
|
assertThat(survey.getDinerCount(), equalTo(2));
|
|
|
|
Diner diner = survey.getDiner(0);
|
|
assertThat(diner.isCategory(0), is(false));
|
|
diner = survey.getDiner(1);
|
|
assertThat(diner.isCategory(0), is(true));
|
|
}
|
|
}
|