opensourceai
diff --git a/‎Dockerfile‎
Lines changed: 1 addition & 0 deletions b/‎Dockerfile‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎tests/test_chainer_chemistry.py‎
Lines changed: 19 additions & 0 deletions b/‎tests/test_chainer_chemistry.py‎
Lines changed: 19 additions & 0 deletions
@@ -489,6 +489,7 @@ RUN pip install flashtext && \
     pip install pykalman && \
     pip install optuna && \
     pip install chainercv && \
+    pip install chainer-chemistry && \
     pip install plotly_express && \
     pip install albumentations && \
     pip install rtree && \
 
@@ -0,0 +1,19 @@
+import numpy
+import unittest
+
+from chainer import cuda
+
+from chainer_chemistry.models.mlp import MLP
+
+
+class TestChainerChemistry(unittest.TestCase):
+
+    def test_mlp(self):
+        batch_size = 2
+        input_dim = 16
+        out_dim = 4
+
+        model = MLP(out_dim=out_dim)
+        data = numpy.random.rand(batch_size, input_dim).astype(numpy.float32)
+        y_actual = cuda.to_cpu(model(data).data)
+        assert y_actual.shape == (batch_size, out_dim)