def test_diff(self):
P1 = NTParameterSet(self.example)
P2 = NTParameterSet({
"y": {
"a": -2,
"b": [4, 5, 6],
"c": 55,
},
"x": 2.9,
"z": 100,
"mylabel": "Dodge City",
"have_horse": True
})
self.assertEqual(P1.diff(P2), ({
'y': {
'c': 5
},
'mylabel': 'camelot',
'have_horse': False
}, {
'y': {
'c': 55
},
'mylabel': 'Dodge City',
'have_horse': True
}))
def test__pop(self):
P = NTParameterSet(self.example)
self.assertEqual(P.pop('x'), 2.9)
self.assertEqual(P.pop('have_horse'), False)
self.assertEqual(P.pop('y'), {"a": -2, "b": [4, 5, 6], "c": 5})
self.assertEqual(P.as_dict(), {'z': 100, 'mylabel': 'camelot'})
self.assertEqual(P.pop('foo', 42), 42)
self.assertEqual(P.pop('foo', None), None)
def test__str(self):
P = NTParameterSet(self.example)
as_string = str(P)
self.assertIsInstance(as_string, str)
self.assertEqual(P, NTParameterSet(as_string))
def test__json_should_be_accepted(self):
P = NTParameterSet(json.dumps(self.example))
self.assertEqual(P.y.a, -2)
self.assertEqual(P.y.b, [4, 5, 6])
self.assertEqual(P.x, 2.9)
self.assertEqual(P.mylabel, "camelot")
