diff --git a/pygal/test/test_config.py b/pygal/test/test_config.py
index ed71224..9c281d1 100644
--- a/pygal/test/test_config.py
+++ b/pygal/test/test_config.py
@@ -18,7 +18,7 @@
 # along with pygal. If not, see <http://www.gnu.org/licenses/>.
 
 from pygal import (
-    Line, Dot, Pie, Radar, Config, Bar, Funnel, Worldmap,
+    Line, Dot, Pie, Treemap, Radar, Config, Bar, Funnel, Worldmap,
     SupranationalWorldmap, Histogram, Gauge, Box, XY,
     Pyramid, DateY, HorizontalBar, HorizontalStackedBar,
     FrenchMap_Regions, FrenchMap_Departments)
@@ -272,7 +272,7 @@ def test_no_data():
 
 def test_include_x_axis(Chart):
     chart = Chart()
-    if Chart in (Pie, Radar, Funnel, Dot, Gauge, Worldmap,
+    if Chart in (Pie, Treemap, Radar, Funnel, Dot, Gauge, Worldmap,
                  SupranationalWorldmap, Histogram, Box,
                  FrenchMap_Regions, FrenchMap_Departments):
         return
@@ -359,7 +359,7 @@ def test_x_y_title(Chart):
 
 def test_x_label_major(Chart):
     if Chart in (
-            Pie, Funnel, Dot, Gauge, Worldmap,
+            Pie, Treemap, Funnel, Dot, Gauge, Worldmap,
             SupranationalWorldmap, Histogram, Box,
             FrenchMap_Regions, FrenchMap_Departments,
             Pyramid, DateY):
@@ -403,7 +403,7 @@ def test_x_label_major(Chart):
 
 def test_y_label_major(Chart):
     if Chart in (
-            Pie, Funnel, Dot, Gauge, Worldmap,
+            Pie, Treemap, Funnel, Dot, Gauge, Worldmap,
             SupranationalWorldmap, Histogram, Box,
             FrenchMap_Regions, FrenchMap_Departments,
             HorizontalBar, HorizontalStackedBar,
diff --git a/pygal/test/test_graph.py b/pygal/test/test_graph.py
index f326f03..2ca2c1a 100644
--- a/pygal/test/test_graph.py
+++ b/pygal/test/test_graph.py
@@ -105,7 +105,7 @@ def test_metadata(Chart):
             'Five', 'http://7.example.com/', 'Seven'):
         assert md in cut(q('desc'), 'text')
 
-    if Chart == pygal.Pie:
+    if Chart in (pygal.Pie, pygal.Treemap):
         # Slices with value 0 are not rendered
         assert len(v) - 1 == len(q('.tooltip-trigger').siblings('.value'))
     elif Chart not in (