Skip to content

Commit

Permalink
Speed up tests a lil
Browse files Browse the repository at this point in the history
  • Loading branch information
GeoArkadeep committed Jan 7, 2025
1 parent c17dbb2 commit 02b467e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/test_package.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
def test_integration_basic():
# Test case 1: Basic well log generation and plotting
attrib = [10, 0, 0, 0, 0, 0, 0, 0]
well = lst.create_random_well(kb=35, gl=-200, step=0.15)
well = lst.create_random_well(kb=35, gl=-200, step=50)
dev = lst.getwelldev(wella=well)
output = lst.plotPPzhang(dev, writeFile=False)

Expand All @@ -21,15 +21,15 @@ def test_integration_basic():
def test_integration_doi():
# Test case 2: Well with attributes and DOI
attrib = [10, -200, 0, 0, 0, 0, 0, 0]
well = lst.create_random_well(kb=312, gl=300, step=1)
well = lst.create_random_well(kb=312, gl=300, step=4)
dev = lst.getwelldev(wella=well, kickoffpoint=3000, final_angle=90, rateofbuild=0.2, azimuth=270)
output = lst.plotPPzhang(dev, attrib=attrib, doi=2625, writeFile=False)
assert output is not None, "plotPPzhang with attributes and DOI did not return an output"

def test_integration_advanced():
# Test case 3: Shallow angle well and custom TECB
attrib = [10, -200, 0, 0, 0, 0, 0, 0]
well = lst.create_random_well(kb=35, gl=-200, step=1)
well = lst.create_random_well(kb=35, gl=-200, step=4)
dev = lst.getwelldev(wella=well, kickoffpoint=300, final_angle=10, rateofbuild=0.2, azimuth=270)
output = lst.plotPPzhang(dev, attrib=attrib, doi=2625, tecb=0.55, writeFile=False)

Expand Down

0 comments on commit 02b467e

Please sign in to comment.