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