def test_concat_merge(self, segy_path, header_index): """Test concat followed by merge.""" s1_before = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="before") s2_before = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="before") s1_after = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="after") s2_after = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="after") index_before = SeismicIndex(s1_before, s2_before, mode="c") index_after = SeismicIndex(s1_after, s2_after, mode="c") _ = SeismicIndex(index_before, index_after, mode="m")
def test_merge_concat(self, segy_path, header_index): """Test merge followed by concat.""" s1_before = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="before") s2_before = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="before") s1_after = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="after") s2_after = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="after") index_s1 = SeismicIndex(s1_before, s1_after, mode="m") index_s2 = SeismicIndex(s2_before, s2_after, mode="m") _ = SeismicIndex(index_s1, index_s2, mode="c")
def test_merge_wrong_index_fails(self, segy_path, header_index): """Merge must fail if surveys are indexed by different headers.""" sur1 = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="before") sur2 = Survey(segy_path, header_index="CDP", header_cols=HEADER_COLS, name="after") with pytest.raises(ValueError): _ = SeismicIndex(sur1, sur2, mode="m")
def test_merge(self, segy_path, header_index): """Test merging of two surveys.""" sur1 = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="before") sur2 = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="after") _ = SeismicIndex(sur1, sur2, mode="m")
def test_merge_wrong_names_fails(self, segy_path, header_index): """Merge must fail if surveys have same names.""" sur1 = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="sur") sur2 = Survey(segy_path, header_index=header_index, header_cols=HEADER_COLS, name="sur") with pytest.raises(ValueError): _ = SeismicIndex(sur1, sur2, mode="m")
def test_concat_wrong_index_fails(self, segy_path, header_index): """Concat must fail if surveys are indexed by different headers.""" sur1 = Survey(segy_path, header_index=header_index, name="sur") sur2 = Survey(segy_path, header_index="CDP", name="sur") with pytest.raises(ValueError): _ = SeismicIndex(sur1, sur2, mode="c")
def test_concat_wrong_names_fails(self, segy_path, header_index): """Concat must fail if surveys have different names.""" sur1 = Survey(segy_path, header_index=header_index, name="sur") sur2 = Survey(segy_path, header_index=header_index, name="not_sur") with pytest.raises(ValueError): _ = SeismicIndex(sur1, sur2, mode="c")
def test_concat(self, segy_path, header_index): """Test concatenation of two surveys.""" sur1 = Survey(segy_path, header_index=header_index, name="sur") sur2 = Survey(segy_path, header_index=header_index, name="sur") _ = SeismicIndex(sur1, sur2, mode="c")
def test_from_index(self, segy_path, header_index): """Test instantiation from an already created index.""" survey = Survey(segy_path, header_index=header_index) index = SeismicIndex(survey) _ = SeismicIndex(index)
def test_from_survey(self, segy_path, header_index): """Test instantiation from a single survey.""" survey = Survey(segy_path, header_index=header_index) _ = SeismicIndex(survey)