def test_references(self):
from bokeh.plotobject import PlotObject
pobject1 = PlotObject()
pobject2 = PlotObject()
pobject3 = PlotObject()
pobject4 = PlotObject()
pobject1.pobject2 = pobject2
pobject1.pobject3 = pobject3
pobject3.pobject4 = pobject4
pobject1.properties_with_refs = Mock(
return_value=['pobject2', 'pobject3'])
pobject3.properties_with_refs = Mock(return_value=['pobject4'])
resultset = set(pobject1.references())
expectedset = set([pobject1, pobject2, pobject3, pobject4])
self.assertEqual(resultset, expectedset)
