Fix a test

theano/d3viz/tests/test_formatting.py

@@ -32,7 +32,7 @@ class TestPyDotFormatter(unittest.TestCase):
         expected = 11
         if th.config.mode == "FAST_COMPILE":
             expected = 12
-        self.assertEqual(len(graph.get_nodes()), 12)
+        self.assertEqual(len(graph.get_nodes()), expected)
         nc = self.node_counts(graph)
 
         if th.config.mode == "FAST_COMPILE":