def test_sub_select(self): related = Related() sub_related = related.sub_select('foo') sub_related.select('bar') sub_related.prefetch('baz') self.assertEqual(related._select, { 'foo__bar' }) self.assertEqual(related._prefetch, { 'foo__baz' })