def test_valid_serialization(self):
# Setup
instance = Edge(0, 2, 5, 'clayton', 1.5)
# Run
result = Edge.from_dict(instance.to_dict())
# Check
assert instance.to_dict() == result.to_dict()
def test_to_dict(self):
"""To_dict returns a dictionary with the parameters to recreate an edge."""
# Setup
edge = Edge(2, 5, 'clayton', 1.5)
edge.D = [1, 3]
expected_result = {
'L': 2,
'R': 5,
'name': 'clayton',
'theta': 1.5,
'D': [1, 3],
'U': None,
'likelihood': None,
'neighbors': None,
'parents': None,
'tau': None
}
# Run
result = edge.to_dict()
# Check
assert result == expected_result