예제 #1
0
    def test_sum(self):
        n1 = symrep.const(1)
        n2 = symrep.const(2)

        n = symrep.sum(n1, n2)
        self.assertEqual(n(0), 3)

        n = symrep.sum(n1, n1, n1)
        self.assertEqual(n(0), 3)
예제 #2
0
 def test_collect_nodes(self):
     n1 = symrep.const(1)
     n2 = symrep.audio.sine(n1)
     n3 = symrep.sum(n1, n2)
     n4 = symrep.const(-2)
     n5 = symrep.product(n3, n4)
     nodes = symrep.collect_nodes(n5)
     self.assertEqual(len(nodes), 5)
     self.assertSetEqual(
         set(n.id for n in nodes),
         set((n1.id, n2.id, n3.id, n4.id, n5.id)))
예제 #3
0
 def test_dot(self):
     n = symrep.product(
         symrep.audio.sine(symrep.const(2)),
         symrep.product(
             symrep.audio.sine(symrep.const(3)),
             symrep.sum(
                 symrep.const(1),
                 symrep.const(1),
             )
         )
     )
     with open("test.dot", "w") as f:
         symrep.to_dot(n, f, name="test_dot")