def testCreateDBLPNode(self): paperDict = {'type': 'Paper', 'id': 68, 'title': 'VLDB Paper 57'} expectedPaper = Paper(id=68, title='VLDB Paper 57') actualPaper = GraphObjectFactory.createDBLPNode(paperDict) self.assertEqual(actualPaper, expectedPaper) authorDict = {'type': 'Author', 'id': 0, 'name': 'Mike'} expectedAuthor = Author(id=0, name='Mike') actualAuthor = GraphObjectFactory.createDBLPNode(authorDict) self.assertEqual(actualAuthor, expectedAuthor) conferenceDict = {'type': 'Conference', 'id': 6, 'name': 'VLDB'} expectedConference = Conference(id=6, name='VLDB') actualConference = GraphObjectFactory.createDBLPNode(conferenceDict) self.assertEqual(actualConference, expectedConference)