#!/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)