Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
fix: fix the unit tests
  • Loading branch information
nan-wang committed Aug 1, 2020
1 parent 06ec237 commit bc2f421
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/unit/flow/test_flow.py
Expand Up @@ -406,8 +406,8 @@ def test_refactor_num_part(self):
self.assertEqual(node.tail_args.socket_out, SocketType.PUB_BIND)

node = f._pod_nodes['r1']
self.assertEqual(node.head_args.socket_in, SocketType.PULL_BIND)
self.assertEqual(node.tail_args.socket_out, SocketType.PUB_BIND)
self.assertEqual(node.head_args.socket_in, SocketType.SUB_CONNECT)
self.assertEqual(node.tail_args.socket_out, SocketType.PUSH_CONNECT)

node = f._pod_nodes['r2']
self.assertEqual(node.head_args.socket_in, SocketType.SUB_CONNECT)
Expand Down

0 comments on commit bc2f421

Please sign in to comment.