# -*- coding: utf-8 -*-

import numpy as np
from inferbeddings.visualization.hinton import HintonDiagram


def test_hinton_diagram():
    data = np.random.randn(10, 50)
    hd = HintonDiagram(is_terminal=False)
    diagram = hd(data)
    assert len(diagram) == 6967

@@ -73,23 +73,16 @@
 tion
-.hinton
  import 
 Hint
@@ -77,23 +77,24 @@
  import 
-H
+h
 inton
-D
+_d
 iagram%0A%0A
@@ -164,62 +164,32 @@
     
-hd = HintonDiagram(is_terminal=False)%0A    
+diagram = hinton_
 diagram
- = hd
 (dat

update tests for hinton diagrams