def test_add_m2_mdict(self):
sample = RockPy.Sample(name='test')
compare = deepcopy(sample.mdict)
m1 = RockPy.Measurement(sample_obj=sample,
mfile=None,
machine='generic',
mtype='mass',
mdata=[1])
m1.add_sval(stype='m1a', sval=1)
m1.add_sval(stype='m1b', sval=2)
sample.add_m2_mdict(mobj=m1)
self.assertTrue(m1 in sample.mdict['stype']['m1a'])
self.assertTrue(m1 in sample.mdict['stype']['m1b'])
m2 = RockPy.Measurement(sample_obj=sample,
mfile=None,
machine='generic',
mtype='height',
mdata=[1])
m2.add_sval(stype='m2a', sval=3)
m2.add_sval(stype='m2b', sval=4)
self.assertTrue(m2 in sample.mdict['stype']['m2a'])
self.assertTrue(m2 in sample.mdict['stype']['m2b'])
m2.add_sval(stype='m2c-added_later', sval=100)
self.assertTrue(m2 in sample.mdict['stype']['m2c-added_later'])
sample.remove_m_from_mdict(mobj=m1)
self.assertFalse('m1a' in sample.mdict['stype'])
self.assertFalse('m1b' in sample.mdict['stype'])
sample.remove_m_from_mdict(mobj=m2)
self.assertDictEqual(compare, sample.mdict)
def test_add_m2_info_dict(self):
sample = RockPy.Sample(name='test')
m = RockPy.Measurement(sample_obj=sample,
mfile=None,
machine='generic',
mtype='mass',
mdata=[1])
sample.add_measurement(mobj=m)
pprint(sample.info_dict)
sample2 = RockPy.Sample(name='test')
sample2.add_m2_info_dict(m)
sample2.add_m2_info_dict(m)
pprint(sample2.info_dict)
