Changed code to create data dict for basis separately.

Basis_Function.py

@@ -44,6 +44,8 @@ class Basis(ABC):
     -------
     calculate_cell_average(projection, stencil_length, add_reconstructions)
         Calculate cell averages for a given projection.
+    create_data_dict()
+        Return dictionary with data necessary to construct basis.
 
     """
 
@@ -207,6 +209,10 @@ class Basis(ABC):
             self.basis)
         return left_reconstructions, right_reconstructions
 
+    def create_data_dict(self):
+        """Return dictionary with data necessary to construct basis."""
+        return {'polynomial_degree': self.polynomial_degree}
+
 
 class Legendre(Basis):
     """Class for Legendre basis."""

Troubled_Cell_Detector.py

@@ -27,6 +27,8 @@ class TroubledCellDetector(ABC):
         Returns string of class name.
     get_cells(projection)
         Calculates troubled cells in a given projection.
+    create_data_dict(projection)
+        Return dictionary with data necessary to plot troubled cells.
 
     """
     def __init__(self, config, init_cond, quadrature, basis, mesh,
@@ -91,7 +93,7 @@ class TroubledCellDetector(ABC):
         """Return dictionary with data necessary to plot troubled cells."""
         return {'projection': projection, 'wave_speed': self._wave_speed,
                 'final_time': self._final_time,
-                'basis': {'polynomial_degree': self._basis.polynomial_degree},
+                'basis': self._basis.create_data_dict(),
                 'mesh': self._mesh.create_data_dict()
                 }