def test_get_key_field(self):
autocomplete = LabeledAutocomplete(model=User)
self.assertEquals('id', autocomplete.get_key_field(User.objects.all()))
autocomplete = LabeledAutocomplete(key_field='id')
self.assertEquals('id', autocomplete.get_key_field(None))
