def test_iter_connections(self):
self.assertEqual(list(Wire('a').iter_wires()), [(None, "a")])
self.assertEqual(list(Bus([Constant(1), Wire('a')]).iter_wires()),
[(1, "a")])
self.assertEqual(list(Bus([Wire('b'), Wire('a')]).iter_wires()),
[(0, "b"), (1, "a")])
self.assertEqual(list(Constant(0).iter_wires()), [])
self.assertEqual(list(NoConnect().iter_wires()), [])
def test_connections(self):
self.assertEqual("a", Wire("a").to_string())
self.assertEqual("1'b0", Constant(0).to_string())
self.assertEqual("1'b1", Constant(1).to_string())
self.assertEqual("{1'b0, 1'b1}",
Bus([Constant(1), Constant(0)]).to_string())
self.assertEqual("{a, 1'b1}",
Bus([Constant(1), Wire('a')]).to_string())
self.assertEqual("", NoConnect().to_string())
def test_rename(self):
self.assertEqual("b", Wire("a").to_string({'a': 'b'}))
self.assertEqual("{b, 1'b1}",
Bus([Constant(1), Wire('a')]).to_string({
'a': 'b'
}))