message: Use regular assertEqual to fix flakiness in Py3.5.
  Prior to 3.6, dict ordering is nondeterministic.
diff: @@ -97,10 +97,10 @@ class StructuredTensorSpecTest(test_util.TensorFlowTestCase,
  ]) # pyformat: disable
  def testSerialize(self, spec, expected):
  serialization = spec._serialize()
  - # TensorShape has an unconventional definition of equality, so we can't use
  - # assertEqual directly here. But repr() is deterministic and lossless for
  - # the expected values, so we can use that instead.
  - self.assertEqual(repr(serialization), repr(expected))
  + # Note that we can only use assertEqual because none of our cases include
  + # a None dimension. A TensorShape with a None dimension is never equal
  + # to another TensorShape.
  + self.assertEqual(serialization, expected)
  @parameterized.parameters([
  (StructuredTensorSpec([1, 2, 3], {}), {}),
  