@Test
  public void classRosterTestAddStudent() {
    Student newStudent = new Student("firstName", "lastName", "studentId", "cohort");

    roster.addStudent(newStudent.getStudentId(), newStudent);

    assertEquals(1, roster.countStudents());
  }
  @Test
  public void classRosterListAllStudents() throws Exception {
    ArrayList<Student> allStudents;

    roster.loadRoster();
    allStudents = roster.listAllStudents();

    assertEquals(0, roster.countStudents());
  }
  @Test
  public void classRosterTestSaveAndLoadRoster() throws Exception {
    Student newStudent = new Student("firstName", "lastName", "studentId", "cohort");

    roster.addStudent(newStudent.getStudentId(), newStudent);
    roster.saveRoster();
    roster.removeStudent("studentId");
    roster.loadRoster();

    assertEquals(1, roster.countStudents());
  }
  @Test
  public void classRosterTestRemoveStudent() {
    roster.removeStudent("studentId");

    assertEquals(0, roster.countStudents());
  }
  @Test
  public void classRosterTestLoadRoster() throws Exception {
    roster.loadRoster();

    assertEquals(1, roster.countStudents());
  }
 @Test
 public void classRosterTestEmptyRoster() {
   assertEquals(0, roster.countStudents());
 }