91f510ce6f4ab0a31824e3fb389155363ce433b3,tests/tests.py,,test_plantcv_white_balance_gray_16bit,#,1187

Before Change

    img = cv2.imread(os.path.join(TEST_DATA, TEST_INPUT_NIR_MASK), -1)
    // Test with debug = "print"
    _ = pcv.white_balance(device=0, img=img, debug="print", roi=(5, 5, 80, 80))
    os.rename("1_whitebalance_roi.png", os.path.join(TEST_TMPDIR, "1_whitebalance_roi.png"))
    os.rename("1_whitebalance.png", os.path.join(TEST_TMPDIR, "1_whitebalance.png"))
    // Test with debug = "plot"
    _ = pcv.white_balance(device=0, img=img, debug="plot", roi=(5, 5, 80, 80))
    // Test without an ROI

After Change

def test_plantcv_white_balance_gray_16bit():
    // Test cache directory
    cache_dir = os.path.join(TEST_TMPDIR, "test_plantcv_white_balance_gray_16bit")
    os.mkdir(cache_dir)
    // Read in test data
    img = cv2.imread(os.path.join(TEST_DATA, TEST_INPUT_NIR_MASK), -1)
    // Test with debug = "print"
    _ = pcv.white_balance(device=0, img=img, debug="print", roi=(5, 5, 80, 80))
    os.rename("1_whitebalance_roi.png", os.path.join(cache_dir, "1_whitebalance_roi.png"))
    os.rename("1_whitebalance.png", os.path.join(cache_dir, "1_whitebalance.png"))
    // Test with debug = "plot"
    _ = pcv.white_balance(device=0, img=img, debug="plot", roi=(5, 5, 80, 80))
    // Test without an ROI
    _ = pcv.white_balance(device=0, img=img, debug=None, roi=None)

Instances