#!/usr/bin/env python
from __future__ import print_function
import unittest
import os
import sys
import matplotlib
import re
from pmagpy import pmag
from pmagpy import ipmag
from pmagpy import new_builder as nb
#from pmagpy import find_pmag_dir
WD = pmag.get_test_WD()
class TestIGRF(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
os.chdir(WD)
def test_igrf_output(self):
result = ipmag.igrf([1999.1, 30, 20, 50])
reference = [1.20288657e+00, 2.82331112e+01, 3.9782338913649881e+04]
for num, item in enumerate(result):
self.assertAlmostEqual(item, reference[num])
class TestUploadMagic(unittest.TestCase):
def setUp(self):
#!/usr/bin/env python
"""
Test map_magic module conversions
from MagIC 2 column names --> MagIC 3 column names.
"""
import unittest
import os
from pmagpy.mapping import map_magic
from pmagpy.mapping import maps
from pmagpy.data_model3 import DataModel
from pmagpy import contribution_builder as cb
from pmagpy import pmag
DM = DataModel()
TEST_DIR = pmag.get_test_WD()
class TestMapping(unittest.TestCase):
def test_meas_map(self):
magic2_keys = {'er_analyst_mail_names', 'measurement_number',
'measurement_flag', 'treatment_temp',
'measurement_pos_z'}
magic2_dict = {key: '' for key in magic2_keys}
magic3_keys = {'analysts', 'measurement',
'quality', 'treat_temp',
'meas_pos_z'}
output = map_magic.mapping(magic2_dict,
map_magic.meas_magic2_2_magic3_map)
# check that translation worked
tests for magic_gui
"""
import wx
import unittest
import os
from programs import magic_gui2 as magic_gui
from pmagpy import builder2 as builder
from pmagpy import pmag
from dialogs import grid_frame2 as grid_frame
from dialogs import pmag_widgets
from pmagpy import validate_upload2 as validate_upload
# set constants
DMODEL = validate_upload.get_data_model()
WD = pmag.get_test_WD() #find_pmag_dir.get_pmag_dir() # sys.prefix
#@unittest.skip('seg fault')
class TestMainFrame(unittest.TestCase):
def setUp(self):
self.app = wx.App()
self.frame = magic_gui.MainFrame(WD, "zebra", DMODEL)
self.app.frame=self.frame
self.pnl = self.frame.GetChildren()[0]
def tearDown(self):
wx.CallAfter(self.frame.Destroy)
wx.CallAfter(self.app.Destroy)
os.chdir(WD)
#!/usr/bin/env python
import unittest
import sys
import os
import wx
from pmagpy import pmag
from dialogs import magic_grid2 as magic_grid
#import wx.lib.inspection
#import numpy as np
#import ipmag
WD = pmag.get_test_WD()
class TestMagicGrid(unittest.TestCase):
"""
testing for MagicGrid class
"""
def setUp(self):
self.app = wx.App()
self.frame = wx.Frame(None, wx.ID_ANY, 'Title', size=(600, 600))
self.frame.pnl = wx.Panel(self.frame, name='a panel')
row_labels = ['alpha', 'bravo', 'charlie', 'whiskey', 'x-ray', 'y', 'z']
col_labels = ['delta', 'echo', 'foxtrot', 'gamma']
self.grid = magic_grid.MagicGrid(self.frame.pnl, 'grid', row_labels, col_labels, size=(600, 600))
self.grid.InitUI()
self.grid.size_grid()
def tearDown(self):
tests for magic_gui
"""
import wx
import unittest
import os
from programs import magic_gui2 as magic_gui
from pmagpy import builder2 as builder
from pmagpy import pmag
from dialogs import grid_frame2 as grid_frame
from dialogs import pmag_widgets
from pmagpy import validate_upload2 as validate_upload
# set constants
DMODEL = validate_upload.get_data_model()
WD = pmag.get_test_WD() #find_pmag_dir.get_pmag_dir() # sys.prefix
#@unittest.skip('seg fault')
class TestMainFrame(unittest.TestCase):
def setUp(self):
self.app = wx.App()
self.frame = magic_gui.MainFrame(WD, "zebra", DMODEL)
self.app.frame = self.frame
self.pnl = self.frame.GetChildren()[0]
def tearDown(self):
wx.CallAfter(self.frame.Destroy)
wx.CallAfter(self.app.Destroy)
os.chdir(WD)
