Remove leftover ipython_version check and add IPython magics test (Closes #407)

Author: fabianp

.github/workflows/lint_python.yml

@@ -74,7 +74,7 @@ jobs:
           python-version: ${{ fromJSON(format('["{0}", "{1}"]', format('{0}.0-alpha - {0}.X', matrix.python.major_dot_minor), matrix.python.major_dot_minor))[startsWith(matrix.python.major_dot_minor, 'pypy')] }}
           architecture: x64
       - run: pip install --upgrade pip wheel
-      - run: pip install pytest
+      - run: pip install pytest ipython
       - run: pip install --editable .
       - run: pip install numpy pylab-sdk
       - run: make test

Makefile

@@ -19,6 +19,7 @@ test:
 	$(PYTHON) test/test_exit_code.py
 	$(PYTHON) test/test_mprof.py
 	$(PYTHON) test/test_async.py
+	$(PYTHON) test/test_ipython.py
 	mprof run test/test_func.py
 
 develop:

memory_profiler.py

@@ -997,10 +997,7 @@ def mprun(self, parameter_s='', cell=None):
         output = stdout_trap.getvalue()
         output = output.rstrip()
 
-        if ipython_version < '0.11':
-            page(output, screen_lines=self.shell.rc.screen_length)
-        else:
-            page(output)
+        page(output)
         print(message, )
 
         text_file = opts.T[0]

test/test_ipython.py

@@ -0,0 +1,37 @@
+import unittest
+from unittest.mock import patch
+
+from IPython.core.interactiveshell import InteractiveShell
+from memory_profiler import LineProfiler, MemitResult, load_ipython_extension
+
+
+def sample_func(n):
+    arr = [1] * n
+    return len(arr)
+
+
+class TestIPythonMagics(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.ip = InteractiveShell.instance()
+        load_ipython_extension(cls.ip)
+        cls.ip.user_ns['sample_func'] = sample_func
+
+    def test_mprun(self):
+        with patch('IPython.core.page.page') as mock_page:
+            prof = self.ip.run_line_magic('mprun', '-r -f sample_func sample_func(1000)')
+        self.assertIsInstance(prof, LineProfiler)
+        self.assertTrue(mock_page.called)
+        output = mock_page.call_args[0][0]
+        self.assertIn('Filename:', output)
+        self.assertIn('def sample_func(n):', output)
+
+    def test_memit(self):
+        res = self.ip.run_line_magic('memit', '-o -q [1] * 10000')
+        self.assertIsInstance(res, MemitResult)
+        self.assertGreater(len(res.mem_usage), 0)
+        self.assertIn('peak memory:', str(res))
+
+
+if __name__ == '__main__':
+    unittest.main()