Changeset 8262

Timestamp:

Dec 5, 2011, 12:16:35 PM (13 years ago)

Author:

steve

Message:

Added in unit test for dump_triangulation

Location:

trunk/anuga_core/source/anuga_parallel

Files:

• parallel_shallow_water.py (modified) (2 diffs)
• test_parallel_shallow_domain.py (added)

trunk/anuga_core/source/anuga_parallel/parallel_shallow_water.py

@@ -22 +22 @@
 #Import matplotlib
 
-import matplotlib
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
-import matplotlib.tri as tri
+
 
 class Parallel_domain(Domain):
@@ -145 +142 @@
     def dump_triangulation(self, filename="domain.png"):
         # Get vertex coordinates, partition full and ghost triangles based on self.tri_full_flag
+
+        try:
+            import matplotlib
+            matplotlib.use('Agg')
+            import matplotlib.pyplot as plt
+            import matplotlib.tri as tri
+        except:
+            print "Couldn't import module from matplotlib, probably you need to update matplotlib"
+            raise
+
         vertices = self.get_vertex_coordinates()
         full_mask = num.repeat(self.tri_full_flag == 1, 3)