def test_product1(self):
got = self._compute_until(self.data, "dense_product_1")
expected = product1_out(sum0_out(product0_out(indicators(self.data))))
self.assertAllClose(got, expected)
def test_root(self):
got = tf.exp(self.discrete_spn(self.data))
expected = root_out(
product1_out(sum0_out(product0_out(indicators(self.data)))))
self.assertAllClose(got, expected)
def test_indicators(self):
got = self._compute_until(self.data, "indicator_leaf")
expected = indicators(self.data)
self.assertAllClose(got, expected)