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)