refactor: safe comparison None (#1604)

jina-ai · Jan 6, 2021 · 62d24c2 · 62d24c2
1 parent 848fbcc
commit 62d24c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/integration/issues/github_1545/test_executor_metas.py b/tests/integration/issues/github_1545/test_executor_metas.py
@@ -7,6 +7,6 @@ def test_separated_workspace(test_metas):
     assert not indexer.separated_workspace
     assert indexer.pea_id == 0
     assert indexer.workspace == './'
-    assert indexer.py_modules == None
+    assert indexer.py_modules is None
     assert indexer.pea_workspace == './/None-0'
     assert indexer.name.startswith('jina.executors.indexers')
diff --git a/tests/unit/executors/indexers/test_binary_indexer.py b/tests/unit/executors/indexers/test_binary_indexer.py
@@ -14,6 +14,7 @@
 @pytest.mark.parametrize('random_workspace_name', ['JINA_TEST_WORKSPACE_BINARY_PB'])
 def test_binarypb_in_flow(test_metas, mocker):
     docs = list(random_docs(10))
+
     def validate(req):
         mock()
         assert len(docs) == len(req.docs)
@@ -23,7 +24,6 @@ def validate(req):
 
     f = Flow(callback_on='body').add(uses='binarypb.yml')
 
-
     with f:
         f.index(docs)
 
@@ -117,8 +117,8 @@ def test_binarypb_delete(test_metas):
         assert idxer.size == 1
 
     with BaseIndexer.load(save_abspath) as idxer:
-        assert idxer.query(1) == None
-        assert idxer.query(2) == None
+        assert idxer.query(1) is None
+        assert idxer.query(2) is None
         assert idxer.query(3) == b'random'