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)