Fix mypy

jingw · Nov 8, 2020 · 052eeee · 052eeee
1 parent 6bd787b
commit 052eeee
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test_pyhdfs.py b/test_pyhdfs.py
@@ -364,7 +364,9 @@ def test_host_in_request(self) -> None:
         with pytest.warns(UserWarning) as record:
             client.get_file_status("foobar://localhost:50070/")
         assert len(record) == 1
-        assert record[0].message.args[0] == "Unexpected scheme foobar"
+        message = record[0].message
+        assert isinstance(message, Warning)
+        assert message.args[0] == "Unexpected scheme foobar"
         assert client.hosts == ["does_not_exist:50070"]
 
     def test_concat(self) -> None: