/tests/imaging/test_plotStrike.py

https://github.com/MTgeophysics/mtpy
Python | 76 lines | 56 code | 14 blank | 6 comment | 3 complexity | 4701d722313b510517f5b3d84f31a32c MD5 | raw file
    

  1. import glob
    2. 

  2. import os
    3. 

  3. 

  4. # configure matplotlib for testing
    5. 

  5. import matplotlib.pyplot as plt
    6. 

  6. 

  7. from mtpy.imaging.plotstrike import PlotStrike
    8. 

  8. from tests.imaging import ImageTestCase, ImageCompare
    9. 

  9. 

  10. edi_paths = [
    11. 

  11.     "data/edifiles",
    12. 

  12.     "examples/data/edi2",
    13. 

  13.     "examples/data/edi_files",
    14. 

  14.     "../MT_Datasets/3D_MT_data_edited_fromDuanJM",
    15. 

  15.     "../MT_Datasets/GA_UA_edited_10s-10000s",
    16. 

  16.     "data/edifiles2"
    17. 

  17. ]
    18. 

  18. 

  19. 

  20. class TestPlotStrike(ImageTestCase):
    21. 

  21.     pass
    22. 

  22. 

  23. 

  24. def _test_gen(edi_path):
    25. 

  25.     def default(self):
    26. 

  26.         edi_file_list = glob.glob(os.path.join(edi_path, "*.edi"))
    27. 

  27.         pt_obj = PlotStrike(fn_list=edi_file_list, plot_yn='n', save_figure_path=self._temp_dir, fig_size=(8, 6), fig_dpi=100)
    28. 

  28.         pt_obj.plot()
    29. 

  29.         plt.pause(1)
    30. 

  30.         save_figure_name = "{}.png".format(default.__name__)
    31. 

  31.         save_figure_path = os.path.join(self._temp_dir, save_figure_name)
    32. 

  32.         pt_obj.save_plot(save_figure_path, file_format='png', close_plot='n')
    33. 

  33.         assert (os.path.isfile(save_figure_path))
    34. 

  34. 

  35.     def rotation(self):
    36. 

  36.         edi_file_list = glob.glob(os.path.join(edi_path, "*.edi"))
    37. 

  37.         # change rotation
    38. 

  38.         pt_obj = PlotStrike(fn_list=edi_file_list, plot_yn='n', rot_z=90, fig_size=(8, 6), fig_dpi=100)
    39. 

  39.         pt_obj.plot()
    40. 

  40.         plt.pause(1)
    41. 

  41. 

  42.     def type(self):
    43. 

  43.         edi_file_list = glob.glob(os.path.join(edi_path, "*.edi"))
    44. 

  44.         # plot type
    45. 

  45.         pt_obj = PlotStrike(fn_list=edi_file_list, plot_yn='n', plot_type=1, fig_size=(8, 6), fig_dpi=100)
    46. 

  46.         pt_obj.plot()
    47. 

  47.         plt.pause(1)
    48. 

  48. 

  49.     def tipper(self):
    50. 

  50.         edi_file_list = glob.glob(os.path.join(edi_path, "*.edi"))
    51. 

  51.         # plot_tipper
    52. 

  52.         pt_obj = PlotStrike(fn_list=edi_file_list, plot_yn='n', plot_tipper='y', fig_size=(8, 6), fig_dpi=100)
    53. 

  53.         pt_obj.plot()
    54. 

  54.         plt.pause(1)
    55. 

  55. 

  56.     def fold(self):
    57. 

  57.         edi_file_list = glob.glob(os.path.join(edi_path, "*.edi"))
    58. 

  58.         # fold
    59. 

  59.         pt_obj = PlotStrike(fn_list=edi_file_list, plot_yn='n', fold=False, fig_size=(8, 6), fig_dpi=100)
    60. 

  60.         pt_obj.plot()
    61. 

  61.         plt.pause(1)
    62. 

  62. 

  63.     return default, rotation, type, tipper, fold
    64. 

  64. 

  65. 

  66. # generate tests
    67. 

  67. for edi_path in edi_paths:
    68. 

  68.     if os.path.isdir(edi_path):
    69. 

  69.         test_name = os.path.basename(edi_path)
    70. 

  70.         for _test_func in _test_gen(edi_path):
    71. 

  71.             _test_func.__name__ = "test_{test_name}_{plot_name}".format(
    72. 

  72.                 test_name=test_name, plot_name=_test_func.__name__)
    73. 

  73.             setattr(
    74. 

  74.                 TestPlotStrike,
    75. 

  75.                 _test_func.__name__,
    76. 

  76.                 ImageCompare(fig_size=(8, 6), savefig_kwargs={"dpi": 100}).__call__(_test_func))
    77.