import { SegmentedToggleUniDriver } from '@wix/design-system/dist/testkit/unidriver';
export function TableGridSwitchButtonUniDriver(base, body) {
    const getSegmented = () => SegmentedToggleUniDriver(base, body);
    return {
        ...getSegmented(),
        clickTableButton: () => base.$$(`[data-hook="toggle-icon"]`).get(1).click(),
        clickGridButton: () => base.$$(`[data-hook="toggle-icon"]`).get(0).click(),
    };
}
export default TableGridSwitchButtonUniDriver;
//# sourceMappingURL=TableGridSwitchButton.uni.driver.js.map