def test_functional_string_empty_SPIN(self):
bqm = dimod.BinaryQuadraticModel.empty(dimod.SPIN)
s = coo.dumps(bqm)
new_bqm = coo.loads(s, dimod.BinaryQuadraticModel.empty(dimod.SPIN))
self.assertEqual(bqm, new_bqm)
def test_dumps_sortable_SPIN(self):
bqm = dimod.BinaryQuadraticModel.from_ising({0: 1.}, {
(0, 1): 2,
(2, 3): .4
})
s = coo.dumps(bqm)
contents = "0 0 1.000000\n0 1 2.000000\n2 3 0.400000"
self.assertEqual(s, contents)
def test_dumps_sortable_SPIN_with_header(self):
bqm = dimod.BinaryQuadraticModel.from_ising({0: 1.}, {
(0, 1): 2,
(2, 3): .4
})
s = coo.dumps(bqm, vartype_header=True)
contents = "# vartype=SPIN\n0 0 1.000000\n0 1 2.000000\n2 3 0.400000"
self.assertEqual(s, contents)
def test_dumps_sortable_BINARY_with_header(self):
bqm = dimod.BinaryQuadraticModel.from_qubo({
(0, 0): 1.,
(0, 1): 2,
(2, 3): .4
})
s = coo.dumps(bqm, vartype_header=True)
contents = "# vartype=BINARY\n0 0 1.000000\n0 1 2.000000\n2 3 0.400000"
self.assertEqual(s, contents)
def test_conflicting_vartype(self):
bqm = dimod.BinaryQuadraticModel({0: 1.}, {
(0, 1): 2,
(2, 3): .4
}, 0.0, dimod.SPIN)
s = coo.dumps(bqm, vartype_header=True)
with self.assertRaises(ValueError):
coo.loads(s, dimod.BinaryQuadraticModel, dimod.BINARY)
def test_no_vartype(self):
bqm = dimod.BinaryQuadraticModel({0: 1.}, {
(0, 1): 2,
(2, 3): .4
}, 0.0, dimod.SPIN)
s = coo.dumps(bqm)
with self.assertRaises(ValueError):
coo.loads(s, dimod.BinaryQuadraticModel)
def test_functional_SPIN_vartypeheader(self):
bqm = dimod.BinaryQuadraticModel({0: 1.}, {
(0, 1): 2,
(2, 3): .4
}, 0.0, dimod.SPIN)
s = coo.dumps(bqm, vartype_header=True)
new_bqm = coo.loads(s, dimod.BinaryQuadraticModel)
self.assertEqual(bqm, new_bqm)
def test_functional_string_SPIN(self):
bqm = dimod.BinaryQuadraticModel({0: 1.}, {
(0, 1): 2,
(2, 3): .4
}, 0.0, dimod.SPIN)
s = coo.dumps(bqm)
new_bqm = coo.loads(s, dimod.BinaryQuadraticModel.empty(dimod.SPIN))
self.assertEqual(bqm, new_bqm)
def test_dumps_empty_SPIN(self):
bqm = dimod.BinaryQuadraticModel.empty(dimod.BINARY)
s = coo.dumps(bqm)
self.assertEqual(s, '')