TestNG是否可以在@BeforeMethod fail上截取屏幕截图？

import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.AfterTest;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.AfterClass;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.apache.commons.io.FileUtils;
import java.io.File;
import java.io.IOException;

public class TestClass {
    private WebDriver driver;

    @BeforeMethod
    public void setUp() {
        // 初始化WebDriver
        System.setProperty("webdriver.chrome.driver", "path/to/chromedriver");
        driver = new ChromeDriver();
    }

    @Test
    public void testMethod() {
        // 执行测试步骤
        driver.get("https://www.example.com");
    }

    @AfterMethod
    public void tearDown(ITestResult result) {
        // 在测试方法执行后进行清理工作
        if (result.getStatus() == ITestResult.FAILURE) {
            // 如果测试失败，则进行屏幕截图
            captureScreenshot(result.getName());
        }
        driver.quit();
    }

    private void captureScreenshot(String testName) {
        // 截取屏幕截图
        File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        try {
            // 将截图保存到指定路径
            FileUtils.copyFile(screenshot, new File("path/to/screenshots/" + testName + ".png"));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}