2 files changed, 42 insertions, 1 deletions
diff --git a/test/fixtures/PAGE.md b/test/fixtures/PAGE.md
index 2839e6d..92ee707 100644
--- a/test/fixtures/PAGE.md
+++ b/test/fixtures/PAGE.md
@@ -30,3 +30,36 @@ Some more nice content ....
 [Cool stuff](http://gitbook.io)
 
 [Link to another Markdown file](./xyz/file.md)
+
+Lets go for another exercise but this time with some context :
+
+---
+
+Exercise with some context code :
+
+Using the `double` function provided, build a `quadruple` function
+
+```py
+
+```
+
+```py
+
+def quadruple(x):
+    return double(double(x))
+
+```
+
+```py
+assert(quadruple(8), 32)
+```
+
+```py
+
+def double(x):
+    return x * 2
+
+```
+
+---
+
diff --git a/test/page.js b/test/page.js
index 0f54809..658559e 100644
--- a/test/page.js
+++ b/test/page.js
@@ -16,7 +16,7 @@ var LINKS_CONTENT = fs.readFileSync(path.join(__dirname, './fixtures/GITHUB_LINK
 
 describe('Page parsing', function() {
     it('should detection sections', function() {
-        assert.equal(LEXED.length, 3);
+        assert.equal(LEXED.length, 4);
     });
 
     it('should detection section types', function() {
@@ -36,6 +36,14 @@ describe('Page parsing', function() {
         assert(LEXED[1].code.base);
         assert(LEXED[1].code.solution);
         assert(LEXED[1].code.validation);
+        assert(LEXED[1].code.context === null);
+
+        assert(LEXED[3].content);
+        assert(LEXED[3].code);
+        assert(LEXED[3].code.base);
+        assert(LEXED[3].code.solution);
+        assert(LEXED[3].code.validation);
+        assert(LEXED[3].code.context);
     });
 
     it('should merge sections correctly', function() {