def test_gen_primary_key_compound_key(self): # gen_primary_key() returns an iterable of values from the model # instance's primary key. storm_object = FakeStormCompoundPrimaryKeyClass() storm_object.id1 = 1234 storm_object.id2 = 5678 self.assertEqual([1234, 5678], list(gen_primary_key(storm_object)))
def test_gen_primary_key(self): # gen_primary_key() returns an iterable of values from the model # instance's primary key. storm_object = FakeStormClass() storm_object.id = 1234 self.assertEqual([1234], list(gen_primary_key(storm_object)))