Label stored results in do_inspect output

Author: hugetim

nbs/03_stata_more.ipynb

@@ -288,7 +288,7 @@
      "text": [
       "var1      __000001\n",
       "var1\n",
-      "Elapsed time: 0.0007 seconds\n"
+      "Elapsed time: 0.0005 seconds\n"
      ]
     }
    ],
@@ -802,8 +802,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Elapsed time: 0.1930 seconds\n",
-      "Elapsed time: 0.0948 seconds\n"
+      "Elapsed time: 0.1942 seconds\n",
+      "Elapsed time: 0.1100 seconds\n"
      ]
     }
    ],
@@ -826,7 +826,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Elapsed time: 0.3255 seconds\n"
+      "Elapsed time: 0.3189 seconds\n"
      ]
     }
    ],
@@ -1124,8 +1124,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Elapsed time: 0.0024 seconds\n",
-      "Elapsed time: 0.2368 seconds\n"
+      "Elapsed time: 0.0023 seconds\n",
+      "Elapsed time: 0.2386 seconds\n"
      ]
     }
    ],
@@ -1261,7 +1261,8 @@
     "#| export\n",
     "def get_inspect(code=\"\", cursor_pos=0, detail_level=0, omit_sections=()):\n",
     "    runner = functools.partial(run_as_program, prog_def_option_code=\"rclass\")\n",
-    "    inspect_code = \"\"\"\\\n",
+    "    inspect_code = \"\"\"\n",
+    "        disp _newline \"*** Stored results:\"\n",
     "        return list\n",
     "        ereturn list\n",
     "        return add\n",
@@ -1270,7 +1271,10 @@
     "        \"\"\"\n",
     "    raw_output = diverted_stata_output(inspect_code, runner=runner)\n",
     "    desc_start = raw_output.find('*** Last updated ')\n",
-    "    return raw_output[desc_start:] + raw_output[:desc_start]"
+    "    out = raw_output[desc_start:]\n",
+    "    if desc_start > 21:\n",
+    "        out += raw_output[:desc_start]\n",
+    "    return out"
    ]
   },
   {
@@ -1342,7 +1346,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "*** Last updated 12:28:02 18 Jan 2023 ***\n",
+      "*** Last updated 15:35:53 23 Jan 2023 ***\n",
       "\n",
       "Contains data\n",
       " Observations:             5                  \n",
@@ -1356,6 +1360,8 @@
       "Sorted by: \n",
       "     Note: Dataset has changed since last saved.\n",
       "\n",
+      "*** Stored results:\n",
+      "\n",
       "scalars:\n",
       "                r(sum) =  2\n",
       "                r(max) =  1\n",
@@ -1387,6 +1393,23 @@
     "run_sfi('return list')"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5f5dc4cb",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| eval: false\n",
+    "run_sfi('clear all')\n",
+    "test_eq(get_inspect()[43:],\"\"\"\\\n",
+    "Contains data\n",
+    " Observations:             0                  \n",
+    "    Variables:             0                  \n",
+    "Sorted by: \n",
+    "\"\"\")"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

nbstata/stata_more.py

@@ -147,7 +147,8 @@ def locals_code_from_dict(preexisting_local_dict):
 # %% ../nbs/03_stata_more.ipynb 73
 def get_inspect(code="", cursor_pos=0, detail_level=0, omit_sections=()):
     runner = functools.partial(run_as_program, prog_def_option_code="rclass")
-    inspect_code = """\
+    inspect_code = """
+        disp _newline "*** Stored results:"
         return list
         ereturn list
         return add
@@ -156,4 +157,7 @@ def get_inspect(code="", cursor_pos=0, detail_level=0, omit_sections=()):
         """
     raw_output = diverted_stata_output(inspect_code, runner=runner)
     desc_start = raw_output.find('*** Last updated ')
-    return raw_output[desc_start:] + raw_output[:desc_start]
+    out = raw_output[desc_start:]
+    if desc_start > 21:
+        out += raw_output[:desc_start]
+    return out