Grid mask support

tests/adapters/test_masking.py

@@ -6,11 +6,7 @@ from datetime import datetime, timedelta
 
 import numpy as np
 
-from finam import (
-    Composition,
-    EsriGrid,
-    Info,
-)
+from finam import Composition, EsriGrid, Info
 from finam.adapters.masking import Masking
 from finam.modules import debug, generators
 
@@ -53,9 +49,9 @@ class TestMasking(unittest.TestCase):
         source.outputs["Output"] >> Masking() >> sink.inputs["Input"]
 
         composition.connect()
+
         self.assertTrue(np.isnan(sink.data["Input"][0][0, 0]))
         self.assertAlmostEqual(sink.data["Input"][0][0, 1].magnitude, 2.0)
-        print(sink.data["Input"][0])
 
 
 if __name__ == "__main__":