Address review

Author: marthacryan

tests/streamlit_ui_tests/grid/keyboard_shortcuts.spec.ts

@@ -54,27 +54,37 @@ test.describe('Keyboard Shortcuts', () => {
 
     test('Next/prev sheet with graphs', async ({ page }) => {
       const mito = await getMitoFrameWithTestCSV(page);
+
+      /* Import another csv and create 2 graphs so that we have enough tabs to navigate between */
       await importCSV(page, mito, 'test.csv');
+      
+      // Create a graph
       await mito.getByText('Graph').click();
       await expect(mito.getByText('Column1 bar chart')).toBeVisible();
+
+      // Create another graph off of test_1
       await mito.locator('.tab', { hasText: 'test_1' }).click();
       await mito.getByText('Graph', { exact: true }).click();
       await expect(mito.getByText('Column1 bar chart')).toBeVisible();
 
+      /* Test navigation */
       await mito.locator('.mito-container').press('Alt+ArrowRight');
-      await awaitResponse(page);
       await expect(mito.locator('.tab-selected')).toHaveText('test');
+
       await mito.locator('.mito-container').press('Alt+ArrowRight');
-      await awaitResponse(page);
       await expect(mito.locator('.tab-selected')).toHaveText('test_1');
 
       await mito.locator('.mito-container').press('Alt+ArrowRight');
-      await awaitResponse(page);
       await expect(mito.locator('.tab-selected')).toHaveText('graph0');
 
       await mito.locator('.mito-container').press('Alt+ArrowLeft');
-      await awaitResponse(page);
       await expect(mito.locator('.tab-selected')).toHaveText('test_1');
+
+      await mito.locator('.mito-container').press('Alt+ArrowLeft');
+      await expect(mito.locator('.tab-selected')).toHaveText('test');
+
+      await mito.locator('.mito-container').press('Alt+ArrowLeft');
+      await expect(mito.locator('.tab-selected')).toHaveText('graph1');
     });
 
     test('Previous Sheet', async ({ page }) => {