public function test_add_supports_custom() { $post_type = rand_str(); $post_type_object = new WP_Post_Type($post_type, array('supports' => array('editor', 'comments', 'revisions'))); $post_type_object->add_supports(); $post_type_supports = get_all_post_type_supports($post_type); $post_type_object->remove_supports(); $post_type_supports_after = get_all_post_type_supports($post_type); $this->assertEqualSets(array('editor' => true, 'comments' => true, 'revisions' => true), $post_type_supports); $this->assertEqualSets(array(), $post_type_supports_after); }