Address review comments

test/EFCore.Cosmos.FunctionalTests/Query/Translations/BasicTypesQueryCosmosFixture.cs

@@ -19,8 +19,18 @@ protected override void OnModelCreating(ModelBuilder modelBuilder, DbContext con
     {
         base.OnModelCreating(modelBuilder, context);
 
-        modelBuilder.Entity<BasicTypesEntity>().ToContainer(nameof(BasicTypesEntity));
-        modelBuilder.Entity<NullableBasicTypesEntity>().ToContainer(nameof(NullableBasicTypesEntity));
+        modelBuilder.Entity<BasicTypesEntity>(
+            builder =>
+            {
+                builder.ToContainer(nameof(BasicTypesEntity));
+                builder.HasPartitionKey(b => b.Id);
+            });
+        modelBuilder.Entity<NullableBasicTypesEntity>(
+            builder =>
+            {
+                builder.ToContainer(nameof(NullableBasicTypesEntity));
+                builder.HasPartitionKey(n => n.Id);
+            });
     }
 
     public Task NoSyncTest(bool async, Func<bool, Task> testCode)

test/EFCore.Specification.Tests/TestModels/BasicTypesModel/BasicFlagsEnum.cs

@@ -9,8 +9,10 @@ public enum BasicFlagsEnum
     Zero = 0,
     One = 1,
     Two = 2,
+    Three = 3,
     Four = 4,
     Eight = 8,
     Sixteen = 16,
     ThirtyTwo = 32,
+    OneTwentyEight = 128,
 }