Add back extra column content

tests/test_gui/test_gui_noveltree.py

@@ -30,8 +30,8 @@ from PyQt6.QtWidgets import QInputDialog, QToolTip
 
 from novelwriter import CONFIG, SHARED
 from novelwriter.dialogs.editlabel import GuiEditLabel
-from novelwriter.enum import nwFocus, nwItemType
-from novelwriter.gui.noveltree import GuiNovelTree, NovelTreeColumn
+from novelwriter.enum import nwFocus, nwItemType, nwNovelExtra
+from novelwriter.gui.noveltree import GuiNovelTree
 from novelwriter.types import QtMouseLeft, QtMouseMiddle
 
 from tests.tools import C, buildTestProject
@@ -143,28 +143,28 @@ def testGuiNovelTree_TreeItems(qtbot, monkeypatch, nwGUI, projPath, mockRnd):
     # Last Column
     # ===========
 
-    novelBar.setLastColType(NovelTreeColumn.HIDDEN)
+    novelBar.setLastColType(nwNovelExtra.HIDDEN)
     assert novelTree.isColumnHidden(novelTree.C_EXTRA) is True
-    assert novelTree.lastColType == NovelTreeColumn.HIDDEN
+    assert novelTree.lastColType == nwNovelExtra.HIDDEN
     assert novelTree._getLastColumnText(C.hSceneDoc, "T0001") == ("", "")
 
-    novelBar.setLastColType(NovelTreeColumn.PLOT)
+    novelBar.setLastColType(nwNovelExtra.PLOT)
     assert novelTree.isColumnHidden(novelTree.C_EXTRA) is False
-    assert novelTree.lastColType == NovelTreeColumn.PLOT
+    assert novelTree.lastColType == nwNovelExtra.PLOT
     assert novelTree._getLastColumnText(C.hSceneDoc, "T0001") == (
         "", ""
     )
 
-    novelBar.setLastColType(NovelTreeColumn.FOCUS)
+    novelBar.setLastColType(nwNovelExtra.FOCUS)
     assert novelTree.isColumnHidden(novelTree.C_EXTRA) is False
-    assert novelTree.lastColType == NovelTreeColumn.FOCUS
+    assert novelTree.lastColType == nwNovelExtra.FOCUS
     assert novelTree._getLastColumnText(C.hSceneDoc, "T0001") == (
         "Jane", "Focus: Jane"
     )
 
-    novelBar.setLastColType(NovelTreeColumn.POV)
+    novelBar.setLastColType(nwNovelExtra.POV)
     assert novelTree.isColumnHidden(novelTree.C_EXTRA) is False
-    assert novelTree.lastColType == NovelTreeColumn.POV
+    assert novelTree.lastColType == nwNovelExtra.POV
     assert novelTree._getLastColumnText(C.hSceneDoc, "T0001") == (
         "Jane", "Point of View: Jane"
     )