public void ReplaceEntitySet_should_remove_set_with_type() { var model = new EdmModel().Initialize(); var entityType = model.AddEntityType("Foo"); model.AddEntitySet("FooSet", entityType); model.ReplaceEntitySet(entityType, null); Assert.Equal(1, model.GetEntityTypes().Count()); Assert.Equal(0, model.Containers.First().EntitySets.Count()); }