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_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)