Escape common magics #35

mwouts · Aug 22, 2018 · 158ed63 · 158ed63
1 parent d5457dc
commit 158ed63
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/nbrmd/magics.py b/nbrmd/magics.py
@@ -20,6 +20,15 @@
                '%timeit  %unalias  %unload_ext  %who  %who_ls  %whos  ' \
                '%xdel  %xmode'.split('  ')
 
+# Add classical line magics
+_LINE_MAGICS += ('%autoreload %aimport '  # autoreload
+                 '%R %Rpush %Rpull %Rget '  # rmagic
+                 '%store '  # storemagic
+                 ).split(' ')
+
+# Remove any blank line
+_LINE_MAGICS = [magic for magic in _LINE_MAGICS if magic.startswith('%')]
+
 # A magic expression is a line or cell magic escaped zero, or multiple times
 _PERCENT_RE = re.compile(r"^(# |#)*%")
 _FORCE_ESC_RE = re.compile(r"^(# |#)*%(.*)(#| )escape")

diff --git a/tests/test_escape_magics.py b/tests/test_escape_magics.py
@@ -3,7 +3,8 @@
 
 
 @pytest.mark.parametrize('line', ['%matplotlib inline', '#%matplotlib inline',
-                                  '##%matplotlib inline', '%%HTML'])
+                                  '##%matplotlib inline', '%%HTML',
+                                  '%autoreload', '%store'])
 def test_escape(line):
     assert escape_magic([line]) == ['# ' + line]
     assert unescape_magic(escape_magic([line])) == [line]