public void ShouldDisplayAdditionalColumns_OnClickOfAdditionalColumnOKButton()
        {
            Select_Year_And_NavigateToEditMarksScreen();
            AttendanceDetails    editMarksGrid     = new AttendanceDetails();
            AdditionalCoulmnPage additionalColumns = editMarksGrid.ClickAdditionalColumn();

            additionalColumns.ClickParentCheckboxes();
            additionalColumns.ClickOkButton();
            // Check the columns are added to the grid
            bool DOBColumn    = SeleniumHelper.FindElement(EditMarksElements.GridColumns.DateOfBirth).Displayed;
            bool genderColumn = SeleniumHelper.FindElement(EditMarksElements.GridColumns.Gender).Displayed;

            Assert.IsTrue(DOBColumn && genderColumn);
        }
        public void ShouldRemoveColumnsinGrid_OnClickOfClearSelection_AdditionalColumnOKButton()
        {
            Select_Year_And_NavigateToEditMarksScreen();
            AttendanceDetails    editMarksGrid     = new AttendanceDetails();
            AdditionalCoulmnPage additionalColumns = editMarksGrid.ClickAdditionalColumn();

            additionalColumns.ClickPersonalDetailsCheckbox();
            AttendanceDetails    editMarksGrid1    = additionalColumns.ClickOkButton();
            AdditionalCoulmnPage additionalColumn1 = editMarksGrid1.ClickAdditionalColumn();

            additionalColumn1.ClearAdditionalColumnSelection();
            additionalColumn1.ClickOkButton();

            IWebElement grid    = SeleniumHelper.Get(By.CssSelector("[data-section-id=\"searchResults\"]"));
            var         columns = grid.FindElements(By.CssSelector("[data-menu-column-id]"));

            // Only the Pupil Name and Session columns should be present in the grid
            Assert.IsTrue(columns.Count == 3);
        }