From c0187ceb3a9ab01209a63c9209a83e1a5bd59b7c Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Mon, 19 Feb 2024 21:01:56 -0800 Subject: [PATCH] Improved test for relation_sizes method --- test/space_test.rb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/space_test.rb b/test/space_test.rb index 24f53de8..551b4fdf 100644 --- a/test/space_test.rb +++ b/test/space_test.rb @@ -7,9 +7,9 @@ def test_database_size def test_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" } + assert relation_sizes.find { |r| r[:relation] == "users" && r[:type] == "table" } + assert relation_sizes.find { |r| r[:relation] == "users_pkey" && r[:type] == "index" } + assert relation_sizes.find { |r| r[:relation] == "all_users" && r[:type] == "matview" } end def test_table_sizes