def test_sac2asdf_failure_cases(tmpdir): tmpdir = tmpdir.strpath output_file = os.path.join(tmpdir, "out.h5") sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append(tmpdir) sys.argv.append(output_file) # Invalid tag. sys.argv.append("RaNdOm") with pytest.raises(ValueError): sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] # Not a folder sys.argv.append(os.path.join(tmpdir, "12345")) sys.argv.append(output_file) sys.argv.append("random") with pytest.raises(ValueError): sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup with io.open(output_file, "wt") as fh: fh.write("1") sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append(tmpdir) # File already exists. sys.argv.append(output_file) sys.argv.append("random") with pytest.raises(ValueError): sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup
def test_sac2asdf_failure_cases(tmpdir): tmpdir = tmpdir.strpath output_file = os.path.join(tmpdir, "out.h5") sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append(tmpdir) sys.argv.append(output_file) # Invalid tag. sys.argv.append("RaNdOm") with pytest.raises(ValueError): sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] # Not a folder sys.argv.append(os.path.join(tmpdir, "12345")) sys.argv.append(output_file) sys.argv.append("random") with pytest.raises(ValueError): sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup with io.open(output_file, "wt") as fh: fh.write("1") sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append(tmpdir) # File already exists. sys.argv.append(output_file) sys.argv.append("random") with pytest.raises(ValueError): sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup
def test_sac2asdf_script(tmpdir, capsys): tmpdir = tmpdir.strpath # Create some test data data_1 = np.arange(10, dtype=np.float32) header = { "kstnm": "ANMO", "knetwk": "IU", "kcmpnm": "BHZ", "stla": 40.5, "stlo": -108.23, "stel": 100.0, "stdp": 3.4, "evla": -15.123, "evlo": 123, "evdp": 50, "nzyear": 2012, "nzjday": 123, "nzhour": 13, "nzmin": 43, "nzsec": 17, "nzmsec": 100, "delta": 1.0 / 40, "o": -10.0, } sac = SACTrace(data=data_1, **header) sac.write(os.path.join(tmpdir, "a.sac")) data_2 = 2.0 * np.arange(10, dtype=np.float32) header = { "kstnm": "BBBB", "knetwk": "AA", "kcmpnm": "CCC", "stla": 40.5, "stlo": -108.23, "stel": 200.0, "stdp": 2.4, "evla": -14.123, "evlo": 125, "evdp": 30, "nzyear": 2013, "nzjday": 123, "nzhour": 13, "nzmin": 43, "nzsec": 17, "nzmsec": 100, "delta": 1.0 / 40, "o": 10.0, } sac = SACTrace(data=data_2, **header) sac.write(os.path.join(tmpdir, "b.sac")) output_file = os.path.join(tmpdir, "out.h5") assert not os.path.exists(output_file) sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append(tmpdir) sys.argv.append(output_file) sys.argv.append("random") sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup non_verbose_out, non_verbose_err = capsys.readouterr() assert not non_verbose_err assert os.path.exists(output_file) with pyasdf.ASDFDataSet(output_file, mode="r") as ds: # 2 Events. assert len(ds.events) == 2 # 2 Stations. assert len(ds.waveforms) == 2 events = ds.events # NOQA # Data should actually be fully identical np.testing.assert_equal(data_1, ds.waveforms.IU_ANMO.random[0].data) np.testing.assert_equal(data_2, ds.waveforms.AA_BBBB.random[0].data) assert ds.waveforms.IU_ANMO.random[0].id == "IU.ANMO..BHZ" assert ds.waveforms.AA_BBBB.random[0].id == "AA.BBBB..CCC" c = ds.waveforms.IU_ANMO.coordinates np.testing.assert_allclose( [c["latitude"], c["longitude"], c["elevation_in_m"]], [40.5, -108.23, 100.0], ) c = ds.waveforms.AA_BBBB.coordinates np.testing.assert_allclose( [c["latitude"], c["longitude"], c["elevation_in_m"]], [40.5, -108.23, 200.0], ) c = ds.waveforms.IU_ANMO.channel_coordinates["IU.ANMO..BHZ"][0] np.testing.assert_allclose( [ c["latitude"], c["longitude"], c["elevation_in_m"], c["local_depth_in_m"], ], [40.5, -108.23, 100.0, 3.4], ) c = ds.waveforms.AA_BBBB.channel_coordinates["AA.BBBB..CCC"][0] np.testing.assert_allclose( [ c["latitude"], c["longitude"], c["elevation_in_m"], c["local_depth_in_m"], ], [40.5, -108.23, 200.0, 2.4], ) # Events origin = (ds.waveforms.IU_ANMO.random[0].stats.asdf.event_ids[0]. get_referred_object().origins[0]) np.testing.assert_allclose( [origin.latitude, origin.longitude, origin.depth], [-15.123, 123.0, 50.0], ) assert (origin.time == obspy.UTCDateTime( year=2012, julday=123, hour=13, minute=43, second=17, microsecond=100000, ) - 10.0) origin = (ds.waveforms.AA_BBBB.random[0].stats.asdf.event_ids[0]. get_referred_object().origins[0]) np.testing.assert_allclose( [origin.latitude, origin.longitude, origin.depth], [-14.123, 125.0, 30.0], ) assert (origin.time == obspy.UTCDateTime( year=2013, julday=123, hour=13, minute=43, second=17, microsecond=100000, ) + 10.0) # Run once again in verbose mode but just test that the output is # actually more. os.remove(output_file) sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append("--verbose") sys.argv.append(tmpdir) sys.argv.append(output_file) sys.argv.append("random") sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup verbose_out, verbose_err = capsys.readouterr() assert not verbose_err assert len(verbose_out) > len(non_verbose_out)
def test_sac2asdf_script(tmpdir, capsys): tmpdir = tmpdir.strpath # Create some test data data_1 = np.arange(10, dtype=np.float32) header = {'kstnm': 'ANMO', 'knetwk': 'IU', 'kcmpnm': 'BHZ', 'stla': 40.5, 'stlo': -108.23, 'stel': 100.0, 'stdp': 3.4, 'evla': -15.123, 'evlo': 123, 'evdp': 50, 'nzyear': 2012, 'nzjday': 123, 'nzhour': 13, 'nzmin': 43, 'nzsec': 17, 'nzmsec': 100, 'delta': 1.0 / 40, 'o': -10.0} sac = SACTrace(data=data_1, **header) sac.write(os.path.join(tmpdir, "a.sac")) data_2 = 2.0 * np.arange(10, dtype=np.float32) header = {'kstnm': 'BBBB', 'knetwk': 'AA', 'kcmpnm': 'CCC', 'stla': 40.5, 'stlo': -108.23, 'stel': 200.0, 'stdp': 2.4, 'evla': -14.123, 'evlo': 125, 'evdp': 30, 'nzyear': 2013, 'nzjday': 123, 'nzhour': 13, 'nzmin': 43, 'nzsec': 17, 'nzmsec': 100, 'delta': 1.0 / 40, 'o': 10.0} sac = SACTrace(data=data_2, **header) sac.write(os.path.join(tmpdir, "b.sac")) output_file = os.path.join(tmpdir, "out.h5") assert not os.path.exists(output_file) sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append(tmpdir) sys.argv.append(output_file) sys.argv.append("random") sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup non_verbose_out, non_verbose_err = capsys.readouterr() assert not non_verbose_err assert os.path.exists(output_file) with pyasdf.ASDFDataSet(output_file, mode="r") as ds: # 2 Events. assert len(ds.events) == 2 # 2 Stations. assert len(ds.waveforms) == 2 events = ds.events # flake8: noqa # Data should actually be fully identical np.testing.assert_equal( data_1, ds.waveforms.IU_ANMO.random[0].data) np.testing.assert_equal( data_2, ds.waveforms.AA_BBBB.random[0].data) assert ds.waveforms.IU_ANMO.random[0].id == "IU.ANMO..BHZ" assert ds.waveforms.AA_BBBB.random[0].id == "AA.BBBB..CCC" c = ds.waveforms.IU_ANMO.coordinates np.testing.assert_allclose( [c["latitude"], c["longitude"], c["elevation_in_m"]], [40.5, -108.23, 100.0]) c = ds.waveforms.AA_BBBB.coordinates np.testing.assert_allclose( [c["latitude"], c["longitude"], c["elevation_in_m"]], [40.5, -108.23, 200.0]) c = ds.waveforms.IU_ANMO.channel_coordinates["IU.ANMO..BHZ"][0] np.testing.assert_allclose( [c["latitude"], c["longitude"], c["elevation_in_m"], c["local_depth_in_m"]], [40.5, -108.23, 100.0, 3.4]) c = ds.waveforms.AA_BBBB.channel_coordinates["AA.BBBB..CCC"][0] np.testing.assert_allclose( [c["latitude"], c["longitude"], c["elevation_in_m"], c["local_depth_in_m"]], [40.5, -108.23, 200.0, 2.4]) # Events origin = ds.waveforms.IU_ANMO.random[ 0].stats.asdf.event_ids[0].get_referred_object().origins[0] np.testing.assert_allclose( [origin.latitude, origin.longitude, origin.depth], [-15.123, 123.0, 50.0]) assert origin.time == obspy.UTCDateTime( year=2012, julday=123, hour=13, minute=43, second=17, microsecond=100000) - 10.0 origin = ds.waveforms.AA_BBBB.random[ 0].stats.asdf.event_ids[0].get_referred_object().origins[0] np.testing.assert_allclose( [origin.latitude, origin.longitude, origin.depth], [-14.123, 125.0, 30.0]) assert origin.time == obspy.UTCDateTime( year=2013, julday=123, hour=13, minute=43, second=17, microsecond=100000) + 10.0 # Run once again in verbose mode but just test that the output is # actually more. os.remove(output_file) sys_argv_backup = copy.copy(sys.argv) try: sys.argv = sys.argv[:1] sys.argv.append("--verbose") sys.argv.append(tmpdir) sys.argv.append(output_file) sys.argv.append("random") sac2asdf.__main__() finally: # Restore to not mess with any of pytests logic. sys.argv = sys_argv_backup verbose_out, verbose_err = capsys.readouterr() assert not verbose_err assert len(verbose_out) > len(non_verbose_out)