Make sock_diag_test work on both python2 and python3. Bug: 243057700 Test: test passes on bramble with both python versions Change-Id: I14c2dc211f022dd90488196f2de9e4eaf3c2a52c