Fix TFF project lint.

PiperOrigin-RevId: 321411362

tensorflow_federated/python/common_libs/tracing_test.py

@@ -221,9 +221,9 @@ class TracingProviderInterfaceTest(absltest.TestCase):
       pass
     self.assertEqual(mock.scopes[0], 'scope')
     self.assertEqual(mock.sub_scopes[0], 'sub_scope')
-    self.assertEqual(mock.parent_span_yields[0], None)
-    self.assertEqual(mock.fn_argss[0], None)
-    self.assertEqual(mock.fn_kwargss[0], None)
+    self.assertIsNone(mock.parent_span_yields[0])
+    self.assertIsNone(mock.fn_argss[0])
+    self.assertIsNone(mock.fn_kwargss[0])
     self.assertEqual(mock.trace_optss[0], {'options': 'some_option'})
     self.assertIsInstance(mock.trace_results[0], tracing.TracedSpan)
 
@@ -263,7 +263,7 @@ class TracingProviderInterfaceTest(absltest.TestCase):
     MyClass.my_func(1, 2, kw=3)
     self.assertEqual(mock.scopes[0], 'MyClass')
     self.assertEqual(mock.sub_scopes[0], 'my_func')
-    self.assertEqual(mock.parent_span_yields[0], None)
+    self.assertIsNone(mock.parent_span_yields[0])
     self.assertEqual(mock.fn_argss[0], (1, 2))
     self.assertEqual(mock.fn_kwargss[0], {'kw': 3})
     self.assertEqual(mock.trace_optss[0], {'options': 'some_option'})

tensorflow_federated/python/core/impl/type_utils_test.py

@@ -27,7 +27,7 @@ from tensorflow_federated.python.core.impl.compiler import building_block_factor
 class TypeUtilsTest(test.TestCase, parameterized.TestCase):
 
   def test_to_canonical_value_with_none(self):
-    self.assertEqual(type_utils.to_canonical_value(None), None)
+    self.assertIsNone(type_utils.to_canonical_value(None))
 
   def test_to_canonical_value_with_int(self):
     self.assertEqual(type_utils.to_canonical_value(1), 1)