diff --git a/docs/source/concepts/ComponentInfo.rst b/docs/source/concepts/ComponentInfo.rst
index 5aa5257c45e70e8cbddc993c29df10203163e9a6..32a3b7fe406a2679bd8147b11ef2fb24773402c0 100644
--- a/docs/source/concepts/ComponentInfo.rst
+++ b/docs/source/concepts/ComponentInfo.rst
@@ -20,15 +20,20 @@ Examples of using ``ComponentInfo`` in python
 
 Print indices of detectors in "bank1" that are masked
 
-.. code-block:: python 
-
-  from mantid.simpleapi import CreateSampleWorkspace
-
-  ws = CreateSampleWorkspace()
-  comp_info = ws.componentInfo()
-  det_info = ws.detectorInfo()
-  bank_index compinfo.indexOfAny('bank1')
-  for det_index in compinfo.detectorsInSubtree(bank_index):
-      if det_info.isMasked(int(det_index)):
-          print('Masked detector of bank1, ', det_index)
-
+.. testcode:: show_masked_detectors_in_bank 
+
+   from mantid.simpleapi import CreateSampleWorkspace
+   
+   ws = CreateSampleWorkspace()
+   comp_info = ws.componentInfo()
+   det_info = ws.detectorInfo()
+   det_info.setMasked(2, True) # Mask  a bank 1 detector for demo
+   det_info.setMasked(len(det_info)-1, True) # Mask detector not in bank 1
+   bank_index = comp_info.indexOfAny('bank1')
+   for det_index in comp_info.detectorsInSubtree(bank_index):
+       if det_info.isMasked(int(det_index)):
+           print('Masked detector index of bank1 is {}'.format(det_index))
+
+.. testoutput:: show_masked_detectors_in_bank
+
+   Masked detector index of bank1 is 2
diff --git a/docs/source/concepts/DetectorInfo.rst b/docs/source/concepts/DetectorInfo.rst
index a7e6783b04c445f992e6565ddc7c6f74a62cd3fb..288e2ca301db14201a675cf31b936d6bf0777a95 100644
--- a/docs/source/concepts/DetectorInfo.rst
+++ b/docs/source/concepts/DetectorInfo.rst
@@ -16,30 +16,26 @@ Introduction
 Python Interface
 ----------------
 
-Examples of using ``DetectorInfo`` in python
+Example of using ``DetectorInfo`` in python
 
-Mask detectors at some distance from the source
+**Mask detectors at some distance from the source**
 
-.. code-block:: python 
 
-  from mantid.simpleapi import CreateSampleWorkspace
+.. testcode:: mask_detectors 
 
-  # Test workspace with instrument
-  ws = CreateSampleWorkspace()
-  det_info = ws.detectorInfo();
-  for item in det_info:
-      if not item.isMonitor and item.l2 > 2.0:
-          item.setMasked(True)
+   from mantid.simpleapi import CreateSampleWorkspace
+   
+   # Test workspace with instrument
+   ws = CreateSampleWorkspace()
+   det_info = ws.detectorInfo();
+   mask_count = 0
+   for item in det_info:
+       if not item.isMonitor and item.l2 > 2.0:
+           item.setMasked(True)
+           mask_count += 1
+   print('masked {} detectors'.format(mask_count))
 
-Print detectors with scattering angle
-
-.. code-block:: python 
-
-  from mantid.simpleapi import CreateSampleWorkspace
-
-  # Test workspace with instrument
-  ws = CreateSampleWorkspace()
-  det_info = ws.detectorInfo()
-  for item in det_info:
-      if item.l2 > 0.01:
-         print(item.index) 
+.. testoutput:: mask_detectors
+  
+   masked 200 detectors
+