respect error type

Author: sentrivana

tests/integrations/spark/test_spark.py

@@ -295,6 +295,9 @@ def stageId(self):  # noqa: N802
             def name(self):
                 return "run-job"
 
+            def attemptId(self):
+                raise Py4JJavaError
+
             def attemptNumber(self):  # noqa: N802
                 return 14
 
@@ -326,6 +329,12 @@ def stageId(self):  # noqa: N802
             def name(self):
                 return "run-job"
 
+            def attemptId(self):
+                raise Py4JJavaError
+
+            def attemptNumber(self):
+                raise Py4JJavaError
+
         class MockStageSubmitted:
             def stageInfo(self):  # noqa: N802
                 stageinf = StageInfo()