lib/cassanity/schema.rb in cassanity-0.6.0.beta5 vs lib/cassanity/schema.rb in cassanity-0.6.0

@@ -1,5 +1,7 @@
+require 'set'
+
 module Cassanity
   class Schema
     # Internal
     attr_reader :primary_keys
 
@@ -55,11 +57,12 @@
       end
     end
 
     # Private
     def primary_keys_are_defined_as_columns?
-      shared_columns = column_names & @primary_keys
-      shared_columns == @primary_keys
+      flattened_primary_keys = @primary_keys.flatten
+      shared_columns = column_names & flattened_primary_keys
+      shared_columns.to_set == flattened_primary_keys.to_set
     end
 
     # Public
     def inspect
       attributes = [