diff --git a/test/space_test.rb b/test/space_test.rb index 827f6aab..24f53de8 100644 --- a/test/space_test.rb +++ b/test/space_test.rb @@ -6,7 +6,10 @@ def test_database_size end def test_relation_sizes - assert database.relation_sizes + relation_sizes = database.relation_sizes + assert relation_sizes.find { |r| r[:relation] == "users" } + assert relation_sizes.find { |r| r[:relation] == "users_pkey" } + assert relation_sizes.find { |r| r[:relation] == "all_users" } end def test_table_sizes