from RockPy.VisualizeV3 import Figure
__author__ = 'mike'
import RockPy
from os.path import join
from RockPy import VisualizeV3
if __name__ == '__main__':
folder = join(RockPy.test_data_path, 'vsm', 'visualizev3_test')
study = RockPy.Study(name='visualize V3 test')
''' SG 1 '''
sg1 = RockPy.SampleGroup(name='SG1')
s1a = RockPy.Sample(name='1a')
s1b = RockPy.Sample(name='1b')
sg1.add_samples([s1a, s1b])
# adding measurements
s1a.add_measurement(mtype='mass', value=62.7, unit='mg')
h1a1 = s1a.add_measurement(mtype='hys', mfile=join(folder, 'FeNi_FeNi20-Jd001\'-G01_HYS_VSM#62,7[mg]_[]_[]##STD014.001'), machine='vsm')
s1a.add_measurement(mtype='mass', value=51.5, unit='mg')
h1a2 = s1a.add_measurement(mtype='hys', mfile=join(folder, 'FeNi_FeNi20-Jd001\'-G02_HYS_VSM#51,5[mg]_[]_[]##STD014.001'), machine='vsm')
s1b.add_measurement(mtype='mass', value=60.0, unit='mg')
h1b1 = s1b.add_measurement(mtype='hys', mfile=join(folder, 'FeNi_FeNi20-Jd120-G02_HYS_VSM#60,0[mg]_[]_[]##STD015.001'), machine='vsm')
s1b.add_measurement(mtype='mass', value=50.8, unit='mg')
h1b2 = s1b.add_measurement(mtype='hys', mfile=join(folder, 'FeNi_FeNi20-Jd120-G03_HYS_VSM#50,8[mg]_[]_[]##STD015.001'), machine='vsm')
# adding series parameters
h1a1.add_sval(stype='mtime', sval=1)
h1a2.add_sval(stype='mtime', sval=1)
def setUp(self):
self.sample_group = RockPy.SampleGroup()
self.sample = RockPy.Sample(name='test_sample')
self.sample_group.add_samples(self.sample)
self.study = RockPy.Study(self.sample_group)