From dfa1576489cb5aed08239a20cd177ea796969181 Mon Sep 17 00:00:00 2001
From: "Bogdan.Kirilenko" <bogdan.kirilenko@jetbrains.com>
Date: Sun, 9 Jun 2024 10:01:27 +0200
Subject: [PATCH] [jupyter] PY-73017 fixes to pass JupyterEditingTest

GitOrigin-RevId: 5b989c12afb4c94a07947db56f8f6e0625c4589c
---
 .../plugins/notebooks/visualization/ui/EditorCellRunButton.kt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellRunButton.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellRunButton.kt
index 647b614b4c1b..c3913fc919ad 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellRunButton.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/ui/EditorCellRunButton.kt
@@ -17,9 +17,9 @@ class EditorCellRunButton(private val editor: EditorEx) {
     if (interval.type != NotebookCellLines.CellType.CODE) return
     val linesRange = interval.lines
 
-    val sourceStartOffset = editor.document.getLineStartOffset(interval.lines.first + 1)
+    val sourceStartOffset = editor.document.getLineEndOffset(interval.lines.first)
     val sourceEndOffset = editor.document.getLineEndOffset(interval.lines.last)
-    if (sourceStartOffset >= sourceEndOffset) return  // PY-72785 don't show for empty cells
+    if (sourceStartOffset + 1 == sourceEndOffset) return  // PY-72785 don't show for empty cells
 
     cellRangeHighlighter = editor.markupModel.addRangeHighlighter(
       sourceStartOffset,