def test_get_inputs_from_pason_rop(self):
"""Tests that setting the inputs from a :class:`PasonData` object works.
"""
pason_data = dripy.PasonData(TEST_PASON_DATA)
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
drill_sim.get_pason_inputs(pason_data, show_plots=False)
self.assertListEqual(drill_sim.pason_inputs['rop'][0][1000:1010],
TEST_EXPECTED_CLEANED_ROP)
def test_drillsim_from_existing_event_file_2019(self):
event_from_file = adripy.event.DrillEvent.read_from_file(
TEST_EVENT_FILE_2019_2)
drill_sim = DrillSim(self.drill_string, event_from_file,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
drill_sim.build()
failures = check_file_contents(
os.path.join(drill_sim.directory, drill_sim.acf_filename),
EXPECTED_DRILLSIM_FROM_EXISTING_EVENT_FILE_ACF_TEXT)
self.assertListEqual(failures, [])
def test_build_evt_contents(self):
"""Tests that the event file created in the DrillSim directory has the correct contents.
"""
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
drill_sim.build()
evt_file = os.path.join(TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME + '.evt')
failures = check_file_contents(evt_file,
EXPECTED_DRILLSIM_EVENT_FILE_TEXT)
self.assertListEqual(failures, [])
def test_write_tiem_orbit_files_directory_contents(self):
"""Tests that DrillSim.directory contains the expected files
"""
# Create a DrillSim object
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
expected_contents = [
drill_sim.event.filename,
os.path.join(drill_sim.directory, drill_sim.string_filename),
drill_sim.solver_settings.filename
]
for tool in drill_sim.string.tools:
expected_contents.append(
os.path.join(drill_sim.directory, tool['Property_File']))
expected_contents.append(
os.path.join(drill_sim.directory,
drill_sim.string.top_drive['Property_File']))
expected_contents.append(
os.path.join(drill_sim.directory,
drill_sim.string.parameters['Hole_Property_File']))
actual_contents = glob.glob(os.path.join(drill_sim.directory, '*'))
self.assertListEqual(sorted(actual_contents),
sorted(expected_contents))
def test_read_from_directory_analysis_built(self):
"""Tests that a DrillSim that is read from a directory that with a simulation that has been built sets the correct value for DrillSim.built.
"""
drill_sim = DrillSim.read_from_directory(
TEST_EXISTING_DRILLSIM_DIRECTORY)
self.assertTrue(drill_sim.built)
def test_modify_solver_settings_before_build(self):
# Load the drill sim
drill_sim = DrillSim.read_from_directory(
TEST_EXISTING_UNBUILT_DRILLSIM_DIRECTORY)
# Backup the drill sim's acf file and ssf files
shutil.copyfile(drill_sim.solver_settings.filename,
drill_sim.solver_settings.filename + '.tmp')
# Read in the new solver settings
new_solver_settings = adripy.DrillSolverSettings.read_from_file(
TEST_SOLVER_SETTINGS_FILE)
# Update the drill sim with the new solver setings
drill_sim.modify_solver_settings(new_solver_settings)
failures = check_file_contents(
drill_sim.solver_settings.filename,
EXPECTED_DRILLSIM_SSF_TEXT_AFTER_SSF_CHANGE)
# Replace files
os.remove(drill_sim.solver_settings.filename)
os.rename(drill_sim.solver_settings.filename + '.tmp',
drill_sim.solver_settings.filename)
# Assert no failures
self.assertListEqual(failures, [])
def test_input_deck_directory_contents(self):
"""Tests that the input deck (adm, acf, cmd files) are written to the directory
"""
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
drill_sim.build()
expected_contents = [
drill_sim.event.filename,
os.path.join(drill_sim.directory, drill_sim.string_filename),
drill_sim.solver_settings.filename,
os.path.join(drill_sim.directory, drill_sim.adm_filename),
os.path.join(drill_sim.directory, drill_sim.acf_filename),
os.path.join(drill_sim.directory, drill_sim.cmd_filename),
os.path.join(drill_sim.directory, 'aview.cmd'),
os.path.join(drill_sim.directory, 'aview.log'),
os.path.join(drill_sim.directory, 'build.cmd'),
]
for tool in drill_sim.string.tools:
tool_file = tool['Property_File']
expected_contents.append(
os.path.join(drill_sim.directory, tool_file))
expected_contents.append(
os.path.join(drill_sim.directory,
drill_sim.string.top_drive['Property_File']))
expected_contents.append(
os.path.join(drill_sim.directory,
drill_sim.string.parameters['Hole_Property_File']))
actual_contents = glob.glob(os.path.join(drill_sim.directory, '*'))
# Remove the aview.cmd.bak file from the actual contents list
if os.path.join(drill_sim.directory,
'aview.cmd.bak') in actual_contents:
actual_contents.remove(
os.path.join(drill_sim.directory, 'aview.cmd.bak'))
# Remove the aview.loq file from the actual contents list
if os.path.join(drill_sim.directory, 'aview.loq') in actual_contents:
actual_contents.remove(
os.path.join(drill_sim.directory, 'aview.loq'))
self.assertListEqual(sorted(actual_contents),
sorted(expected_contents))
def test_read_from_directory_analysis_name(self):
"""Tests that a DrillSim that is read from a directory has the correct analysis name.
"""
drill_sim = DrillSim.read_from_directory(
TEST_EXISTING_DRILLSIM_DIRECTORY)
self.assertEqual(drill_sim.analysis_name,
TEST_EXISTING_DRILLSIM_ANALYSIS_NAME)
def test_stabilizer_measure_on_object(self):
"""Tests that measure is turned off on drill pipe in the string file.
"""
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
measure = drill_sim.string.tools[1]['Measure']
self.assertEqual(measure, 'yes')
def test_relativity_in_string_event_reference(self):
"""Tests that the file references in the string file are relative to drill_sim.directory
"""
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
event_filepath = adripy.get_TO_param(
os.path.join(drill_sim.directory, drill_sim.string_filename),
'Event_Property_File')
self.assertFalse(
thornpy.utilities.convert_path(drill_sim.directory) in
thornpy.utilities.convert_path(event_filepath))
def test_write_tiem_orbit_files_ssf_filename(self):
"""Tests that DrillSim.event has the correct string filename
"""
# Create a DrillSim object
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
# Test the filename
expected_filename = os.path.join(TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME + '.ssf')
actual_filename = drill_sim.solver_settings.filename
self.assertEqual(actual_filename, expected_filename)
def test_string_file_contents(self):
"""Checks that the contents of the string file generated by drill_sim.setup() are as expected.
"""
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
failures = check_file_contents(
os.path.join(drill_sim.directory, drill_sim.string_filename),
EXPECTED_DRILLSIM_STRING_FILE_TEXT)
self.assertListEqual(failures, [])
def setUp(self):
# Create a test config file containing the test database
adripy.create_cfg_file(TEST_CONFIG_FILENAME,
[TEST_DATABASE_PATH, TEST_NEW_DATABASE_PATH])
# Create the res and msg files
for file in glob.glob(
os.path.join(TEST_EXISTING_DRILLSIM_DIRECTORY, '*.cached')):
shutil.copyfile(file, file.replace('.cached', ''))
self.drill_sim = DrillSim.read_from_directory(
TEST_EXISTING_DRILLSIM_DIRECTORY)
if not self.drill_sim.solved:
self.drill_sim.run()
def test_relativity_in_string_pdc_reference(self):
"""Tests that the file references in the string file are relative to drill_sim.directory
"""
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
_name, pdc_filepath, _so, _gn = adripy.get_tool_name(
os.path.join(drill_sim.directory, drill_sim.string_filename),
'pdc_bit',
return_full_path=True)
self.assertFalse(
thornpy.utilities.convert_path(drill_sim.directory) in
thornpy.utilities.convert_path(pdc_filepath))
def test_read_from_file_then_build(self):
"""Tests that a DrillSim can be built after being read in from a directory.
"""
# Read in the DrillSim
drill_sim = DrillSim.read_from_directory(
TEST_EXISTING_UNBUILT_DRILLSIM_DIRECTORY)
# Build the drill_sim
drill_sim.build()
# Assert that the acf file exists
self.assertTrue(
os.path.exists(
os.path.join(drill_sim.directory, drill_sim.acf_filename)))
def test_clean_pason_data_rop(self):
"""Tests that :meth:`clean_pason` returns the expected values for the rop signal
"""
pason_data = dripy.PasonData(TEST_PASON_DATA)
signal, _time = DrillSim.clean_pason(pason_data,
'rop',
show_plot=False)
# # Plot for confirmation
# pason_data.plt.plot(pason_data.time, pason_data.rop)
# pason_data.plt.plot(_time, signal)
# pason_data.plt.show()
self.assertListEqual(signal[1000:1010], TEST_EXPECTED_CLEANED_ROP)
def test_read_shrink_read(self):
"""Tests that drill_sim results can be read-and-shrunk, then read again.
"""
# Read and shrink
self.drill_sim.read_results(t_min=80,
t_max=99,
shrink_results_file=True)
old_reqs = [k for k in self.drill_sim.results]
new_drill_sim = DrillSim.read_from_directory(
TEST_EXISTING_DRILLSIM_DIRECTORY)
new_drill_sim.read_results(t_min=80, t_max=99)
new_reqs = [k for k in new_drill_sim.results]
self.assertEqual(sorted(old_reqs), sorted(new_reqs))
def test_build_adm_contents(self):
"""Tests that the event file created in the DrillSim directory has the correct contents.
"""
drill_sim = DrillSim(self.drill_string, self.event,
self.solver_settings, TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME)
pason_data = dripy.PasonData(TEST_PASON_DATA)
drill_sim.get_pason_inputs(pason_data, show_plots=False)
drill_sim.build()
adm_file = os.path.join(TEST_WORKING_DIRECTORY,
TEST_ANALYSIS_NAME + '.adm')
n_adm_lines = len(open(adm_file, 'r').readlines())
expected_n_adm_lines = 12538
self.assertEqual(n_adm_lines, expected_n_adm_lines)
def setUp(self):
# Create a test config file containing the test database
adripy.create_cfg_file(TEST_CONFIG_FILENAME,
[TEST_DATABASE_PATH, TEST_NEW_DATABASE_PATH])
# Create a DrillTool object representing a stabilizer
self.pdc_bit = adripy.DrillTool(TEST_PDC_FILE)
# Create a DrillTool object representing a stabilizer
self.stabilizer = adripy.DrillTool(TEST_STABILIZER_FILE)
# Create a DrillTool object representing a drill pipe
self.drill_pipe = adripy.DrillTool(TEST_DRILLPIPE_FILE)
# Create a DrillTool object representing EUS
self.eus = adripy.DrillTool(TEST_EUS_FILE)
# Create a DrillTool object representing a top drive
self.top_drive = adripy.DrillTool(TEST_TOP_DRIVE_FILE)
# Create a DrillString object
self.drill_string = adripy.DrillString(TEST_STRING_NAME,
TEST_EXISTING_HOLE_FILE,
TEST_EVENT_FILE)
# Add the DrillTool objects to the DrillString object
self.drill_string.add_tool(self.pdc_bit, measure='yes')
self.drill_string.add_tool(self.stabilizer, measure='yes')
self.drill_string.add_tool(self.drill_pipe,
joints=20,
group_name='Upper_DP_Group')
self.drill_string.add_tool(self.eus,
joints=20,
group_name='equivalent_pipe',
equivalent=True)
self.drill_string.add_tool(self.top_drive)
# Create an event object
self.event = adripy.DrillEvent(TEST_EVENT_NAME, 2000, 3)
self.event.add_simulation_step(10)
self.event.add_simulation_step(100)
self.event.add_ramp('FLOW_RATE', 0, 15, 500)
self.event.add_ramp('ROTARY_RPM', 15, 15, 60)
self.event.add_ramp('WOB', 30, 15, 50000)
self.event.add_ramp('ROP', 30, 15, 100)
# Create a solver settings object
self.solver_settings = adripy.DrillSolverSettings(
TEST_SOLVER_SETTINGS_NAME)
# If the "existing" drill sims don't exit, create them
for directory in [
TEST_EXISTING_DRILLSIM_DIRECTORY,
TEST_EXISTING_UNBUILT_DRILLSIM_DIRECTORY
]:
if not os.path.exists(directory):
create_existing_drillsim(directory)
drill_sim = DrillSim.read_from_directory(
TEST_EXISTING_DRILLSIM_DIRECTORY)
if drill_sim.built is False:
drill_sim.build()
