Extend the previous test to cover both line and cell modes.

IPython/core/tests/test_magic.py

@@ -450,8 +450,8 b' def test_timeit_arguments():'
     _ip.magic("timeit ('#')")
 
 
-def test_timeit_cell():
-    "Test %%timeit called in cell mode"
+def test_timeit_special_syntax():
+    "Test %%timeit with IPython special syntax"
     from IPython.core.magic import register_line_magic
 
     @register_line_magic
@@ -459,8 +459,12 b' def test_timeit_cell():'
         ip = get_ipython()
         ip.user_ns['lmagic_out'] = line
 
-    _ip.run_cell_magic('timeit', '-n1 -r1', '%lmagic my line')
+    # line mode test
+    _ip.run_line_magic('timeit', '-n1 -r1 %lmagic my line')
     nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line')
+    # cell mode test
+    _ip.run_cell_magic('timeit', '-n1 -r1', '%lmagic my line2')
+    nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line2')
     
 
 @dec.skipif(execution.profile is None)