diff --git a/tests/test_connector.py b/tests/test_connector.py index 36df6f72f08ac0a95f7e7c6f32649b5fd53a3075..40041199cf6cad3df29273c7535d04af081d349e 100644 --- a/tests/test_connector.py +++ b/tests/test_connector.py @@ -1,5 +1,5 @@ import unittest -from pafsim.connector import Connector, ConnectionError +from pafsim.connector import Connector, EdgeConnectionError import pafsim.processor as pp class TEST_Connector(unittest.TestCase): @@ -14,6 +14,6 @@ class TEST_Connector(unittest.TestCase): proc1 = pp.Correlator("dummy1") proc2 = pp.Correlator("dummy2") conn = Connector(proc1, proc2) - with self.assertRaises(ConnectionError): + with self.assertRaises(EdgeConnectionError): conn.connect()