I am using fixture to load testing data using ndb for my Google App Engine project using this patch. Inside my unit tests I often need to reference the underlying entity which causes me to do things like:
# MyEntityClass is an ndb.Model subclass
# MyEntityClassData is a fixture.DataSet
entity = MyEntityClass.get_by_id(MyEntityClassData.entity_one.id_field)
This just feels dirty. Is there a way to extract the entity class created for MyEntityClassData.entity_one
directly from the fixture class?