hydrus/include/TestHydrusTags.py

895 lines
50 KiB
Python

import collections
import HydrusConstants as HC
import HydrusTags
import os
import TestConstants
import unittest
class TestMergeTagsManagers( unittest.TestCase ):
def test_merge( self ):
first = os.urandom( 32 )
second = os.urandom( 32 )
third = os.urandom( 32 )
#
service_keys_to_statuses_to_tags = collections.defaultdict( HC.default_dict_set )
service_keys_to_statuses_to_tags[ first ][ HC.CURRENT ] = { 'current_1', 'series:blame!' }
service_keys_to_statuses_to_tags[ second ][ HC.CURRENT ] = { 'current_duplicate_1', 'character:cibo' }
service_keys_to_statuses_to_tags[ second ][ HC.DELETED ] = { 'current_1' }
service_keys_to_statuses_to_tags[ second ][ HC.PENDING ] = { 'pending_1', 'creator:tsutomu nihei' }
service_keys_to_statuses_to_tags[ second ][ HC.PETITIONED ] = { 'petitioned_1' }
service_keys_to_statuses_to_tags[ third ][ HC.CURRENT ] = { 'current_duplicate', 'current_duplicate_1' }
service_keys_to_statuses_to_tags[ third ][ HC.PENDING ] = { 'volume:3' }
tags_manager_1 = HydrusTags.TagsManager( service_keys_to_statuses_to_tags )
tags_manager_1._RecalcCombined()
#
service_keys_to_statuses_to_tags = collections.defaultdict( HC.default_dict_set )
service_keys_to_statuses_to_tags[ first ][ HC.CURRENT ] = { 'current_2', 'series:blame!', 'chapter:1' }
service_keys_to_statuses_to_tags[ first ][ HC.DELETED ] = { 'deleted_2' }
service_keys_to_statuses_to_tags[ second ][ HC.CURRENT ] = { 'current_duplicate' }
service_keys_to_statuses_to_tags[ second ][ HC.PENDING ] = { 'architecture', 'chapter:2' }
service_keys_to_statuses_to_tags[ third ][ HC.CURRENT ] = { 'current_duplicate' }
tags_manager_2 = HydrusTags.TagsManager( service_keys_to_statuses_to_tags )
tags_manager_2._RecalcCombined()
#
service_keys_to_statuses_to_tags = collections.defaultdict( HC.default_dict_set )
service_keys_to_statuses_to_tags[ second ][ HC.CURRENT ] = { 'page:4', 'page:5' }
service_keys_to_statuses_to_tags[ second ][ HC.PENDING ] = { 'title:double page spread' }
tags_manager_3 = HydrusTags.TagsManager( service_keys_to_statuses_to_tags )
tags_manager_3._RecalcCombined()
#
tags_managers = ( tags_manager_1, tags_manager_2, tags_manager_3 )
tags_manager = HydrusTags.MergeTagsManagers( tags_managers )
#
result = { 'creator' : { 'tsutomu nihei' }, 'series' : { 'blame!' }, 'title' : { 'double page spread' }, 'volume' : { '3' }, 'chapter' : { '1', '2' }, 'page' : { '4', '5' } }
self.assertEqual( tags_manager.GetCombinedNamespaces( ( 'creator', 'series', 'title', 'volume', 'chapter', 'page' ) ), result )
self.assertEqual( tags_manager.GetNamespaceSlice( ( 'character', ) ), frozenset( { 'character:cibo' } ) )
class TestTagsManager( unittest.TestCase ):
@classmethod
def setUpClass( self ):
self._first_key = os.urandom( 32 )
self._second_key = os.urandom( 32 )
self._third_key = os.urandom( 32 )
service_keys_to_statuses_to_tags = collections.defaultdict( HC.default_dict_set )
service_keys_to_statuses_to_tags[ self._first_key ][ HC.CURRENT ] = { 'current', u'\u2835', 'creator:tsutomu nihei', 'series:blame!', 'title:test title', 'volume:3', 'chapter:2', 'page:1' }
service_keys_to_statuses_to_tags[ self._first_key ][ HC.DELETED ] = { 'deleted' }
service_keys_to_statuses_to_tags[ self._second_key ][ HC.CURRENT ] = { 'deleted', u'\u2835' }
service_keys_to_statuses_to_tags[ self._second_key ][ HC.DELETED ] = { 'current' }
service_keys_to_statuses_to_tags[ self._second_key ][ HC.PENDING ] = { 'pending' }
service_keys_to_statuses_to_tags[ self._second_key ][ HC.PETITIONED ] = { 'petitioned' }
service_keys_to_statuses_to_tags[ self._third_key ][ HC.CURRENT ] = { 'petitioned' }
service_keys_to_statuses_to_tags[ self._third_key ][ HC.DELETED ] = { 'pending' }
self._tags_manager = HydrusTags.TagsManager( service_keys_to_statuses_to_tags )
self._tags_manager._RecalcCombined()
self._service_keys_to_statuses_to_tags = service_keys_to_statuses_to_tags
#
self._pending_service_key = os.urandom( 32 )
self._content_update_service_key = os.urandom( 32 )
self._reset_service_key = os.urandom( 32 )
other_service_keys_to_statuses_to_tags = collections.defaultdict( HC.default_dict_set )
other_service_keys_to_statuses_to_tags[ self._pending_service_key ][ HC.PENDING ] = { 'pending' }
other_service_keys_to_statuses_to_tags[ self._pending_service_key ][ HC.PETITIONED ] = { 'petitioned' }
other_service_keys_to_statuses_to_tags[ self._reset_service_key ][ HC.CURRENT ] = { 'reset_current' }
other_service_keys_to_statuses_to_tags[ self._reset_service_key ][ HC.DELETED ] = { 'reset_deleted' }
other_service_keys_to_statuses_to_tags[ self._reset_service_key ][ HC.PENDING ] = { 'reset_pending' }
other_service_keys_to_statuses_to_tags[ self._reset_service_key ][ HC.PETITIONED ] = { 'reset_petitioned' }
self._other_tags_manager = HydrusTags.TagsManager( other_service_keys_to_statuses_to_tags )
self._other_tags_manager._RecalcCombined()
self._other_service_keys_to_statuses_to_tags = other_service_keys_to_statuses_to_tags
def test_get_cstvcp( self ):
result = { 'creator' : { 'tsutomu nihei' }, 'series' : { 'blame!' }, 'title' : { 'test title' }, 'volume' : { '3' }, 'chapter' : { '2' }, 'page' : { '1' } }
self.assertEqual( self._tags_manager.GetCombinedNamespaces( ( 'creator', 'series', 'title', 'volume', 'chapter', 'page' ) ), result )
def test_delete_pending( self ):
self.assertEqual( self._other_tags_manager.GetPending( self._pending_service_key ), { 'pending' } )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._pending_service_key ), { 'petitioned' } )
self._other_tags_manager.DeletePending( self._pending_service_key )
self.assertEqual( self._other_tags_manager.GetPending( self._pending_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._pending_service_key ), set() )
def test_get_current( self ):
self.assertEqual( self._tags_manager.GetCurrent( self._first_key ), { 'current', u'\u2835', 'creator:tsutomu nihei', 'series:blame!', 'title:test title', 'volume:3', 'chapter:2', 'page:1' } )
self.assertEqual( self._tags_manager.GetCurrent( self._second_key ), { 'deleted', u'\u2835' } )
self.assertEqual( self._tags_manager.GetCurrent( self._third_key ), { 'petitioned' } )
self.assertEqual( self._tags_manager.GetCurrent(), { 'current', 'deleted', u'\u2835', 'creator:tsutomu nihei', 'series:blame!', 'title:test title', 'volume:3', 'chapter:2', 'page:1', 'petitioned' } )
def test_get_deleted( self ):
self.assertEqual( self._tags_manager.GetDeleted( self._first_key ), { 'deleted' } )
self.assertEqual( self._tags_manager.GetDeleted( self._second_key ), { 'current' } )
self.assertEqual( self._tags_manager.GetDeleted( self._third_key ), { 'pending' } )
self.assertEqual( self._tags_manager.GetDeleted(), set() ) # combined tag service does not track deleted
def test_get_namespace_slice( self ):
self.assertEqual( self._tags_manager.GetNamespaceSlice( ( 'creator', 'series' ), collapse_siblings = False ), frozenset( { 'creator:tsutomu nihei', 'series:blame!' } ) )
self.assertEqual( self._tags_manager.GetNamespaceSlice( (), collapse_siblings = False ), frozenset() )
self.assertEqual( self._tags_manager.GetNamespaceSlice( ( 'creator', 'series' ) ), frozenset( { 'creator:tsutomu nihei', 'series:blame!' } ) )
self.assertEqual( self._tags_manager.GetNamespaceSlice( () ), frozenset() )
def test_get_num_tags( self ):
self.assertEqual( self._tags_manager.GetNumTags( self._first_key, include_current_tags = False, include_pending_tags = False ), 0 )
self.assertEqual( self._tags_manager.GetNumTags( self._first_key, include_current_tags = True, include_pending_tags = False ), 8 )
self.assertEqual( self._tags_manager.GetNumTags( self._first_key, include_current_tags = False, include_pending_tags = True ), 0 )
self.assertEqual( self._tags_manager.GetNumTags( self._first_key, include_current_tags = True, include_pending_tags = True ), 8 )
self.assertEqual( self._tags_manager.GetNumTags( self._second_key, include_current_tags = False, include_pending_tags = False ), 0 )
self.assertEqual( self._tags_manager.GetNumTags( self._second_key, include_current_tags = True, include_pending_tags = False ), 2 )
self.assertEqual( self._tags_manager.GetNumTags( self._second_key, include_current_tags = False, include_pending_tags = True ), 1 )
self.assertEqual( self._tags_manager.GetNumTags( self._second_key, include_current_tags = True, include_pending_tags = True ), 3 )
self.assertEqual( self._tags_manager.GetNumTags( self._third_key, include_current_tags = False, include_pending_tags = False ), 0 )
self.assertEqual( self._tags_manager.GetNumTags( self._third_key, include_current_tags = True, include_pending_tags = False ), 1 )
self.assertEqual( self._tags_manager.GetNumTags( self._third_key, include_current_tags = False, include_pending_tags = True ), 0 )
self.assertEqual( self._tags_manager.GetNumTags( self._third_key, include_current_tags = True, include_pending_tags = True ), 1 )
self.assertEqual( self._tags_manager.GetNumTags( HC.COMBINED_TAG_SERVICE_KEY, include_current_tags = False, include_pending_tags = False ), 0 )
self.assertEqual( self._tags_manager.GetNumTags( HC.COMBINED_TAG_SERVICE_KEY, include_current_tags = True, include_pending_tags = False ), 10 )
self.assertEqual( self._tags_manager.GetNumTags( HC.COMBINED_TAG_SERVICE_KEY, include_current_tags = False, include_pending_tags = True ), 1 )
self.assertEqual( self._tags_manager.GetNumTags( HC.COMBINED_TAG_SERVICE_KEY, include_current_tags = True, include_pending_tags = True ), 11 )
def test_get_pending( self ):
self.assertEqual( self._tags_manager.GetPending( self._first_key ), set() )
self.assertEqual( self._tags_manager.GetPending( self._second_key ), { 'pending' } )
self.assertEqual( self._tags_manager.GetPending( self._third_key ), set() )
self.assertEqual( self._tags_manager.GetPending(), { 'pending' } )
def test_get_petitioned( self ):
self.assertEqual( self._tags_manager.GetPetitioned( self._first_key ), set() )
self.assertEqual( self._tags_manager.GetPetitioned( self._second_key ), { 'petitioned' } )
self.assertEqual( self._tags_manager.GetPetitioned( self._third_key ), set() )
self.assertEqual( self._tags_manager.GetPetitioned(), set() ) # combined tag service does not track petitioned
def test_get_service_keys_to_statuses_to_tags( self ):
s = self._tags_manager.GetServiceKeysToStatusesToTags()
self.assertEqual( s[ self._first_key ], self._service_keys_to_statuses_to_tags[ self._first_key ] )
self.assertEqual( s[ self._second_key ], self._service_keys_to_statuses_to_tags[ self._second_key ] )
self.assertEqual( s[ self._third_key ], self._service_keys_to_statuses_to_tags[ self._third_key ] )
def test_get_statuses_to_tags( self ):
self.assertEqual( self._tags_manager.GetStatusesToTags( self._first_key ), self._service_keys_to_statuses_to_tags[ self._first_key ] )
self.assertEqual( self._tags_manager.GetStatusesToTags( self._second_key ), self._service_keys_to_statuses_to_tags[ self._second_key ] )
self.assertEqual( self._tags_manager.GetStatusesToTags( self._third_key ), self._service_keys_to_statuses_to_tags[ self._third_key ] )
def test_has_tag( self ):
self.assertTrue( self._tags_manager.HasTag( u'\u2835' ) )
self.assertFalse( self._tags_manager.HasTag( 'not_exist' ) )
def test_process_content_update( self ):
hashes = { os.urandom( 32 ) for i in range( 6 ) }
#
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertNotIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_DELETE, ( 'hello', hashes ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertNotIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_PENDING, ( 'hello', hashes ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertNotIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_RESCIND_PENDING, ( 'hello', hashes ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertNotIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_PENDING, ( 'hello', hashes ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertNotIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_ADD, ( 'hello', hashes ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_PETITION, ( 'hello', hashes, 'reason' ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), { 'hello' } )
self.assertIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_RESCIND_PETITION, ( 'hello', hashes ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_PETITION, ( 'hello', hashes, 'reason' ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), { 'hello' } )
self.assertIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
#
content_update = HC.ContentUpdate( HC.CONTENT_DATA_TYPE_MAPPINGS, HC.CONTENT_UPDATE_DELETE, ( 'hello', hashes ) )
self._other_tags_manager.ProcessContentUpdate( self._content_update_service_key, content_update )
self.assertEqual( self._other_tags_manager.GetCurrent( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetDeleted( self._content_update_service_key ), { 'hello' } )
self.assertEqual( self._other_tags_manager.GetPending( self._content_update_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._content_update_service_key ), set() )
self.assertNotIn( 'hello', self._other_tags_manager.GetCurrent() )
self.assertNotIn( 'hello', self._other_tags_manager.GetPending() )
def test_reset_service( self ):
self.assertEqual( self._other_tags_manager.GetCurrent( self._reset_service_key ), { 'reset_current' } )
self.assertEqual( self._other_tags_manager.GetDeleted( self._reset_service_key ), { 'reset_deleted' } )
self.assertEqual( self._other_tags_manager.GetPending( self._reset_service_key ), { 'reset_pending' } )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._reset_service_key ), { 'reset_petitioned' } )
self._other_tags_manager.ResetService( self._reset_service_key )
self.assertEqual( self._other_tags_manager.GetCurrent( self._reset_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetDeleted( self._reset_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPending( self._reset_service_key ), set() )
self.assertEqual( self._other_tags_manager.GetPetitioned( self._reset_service_key ), set() )
class TestTagObjects( unittest.TestCase ):
def test_predicates( self ):
p = HC.Predicate( HC.PREDICATE_TYPE_TAG, 'tag' )
self.assertEqual( p.GetUnicode(), u'tag' )
p = HC.Predicate( HC.PREDICATE_TYPE_TAG, 'tag', counts = { HC.CURRENT : 1, HC.PENDING : 2 } )
self.assertEqual( p.GetUnicode( with_count = False ), u'tag' )
self.assertEqual( p.GetUnicode( with_count = True ), u'tag (1) (+2)' )
p = HC.Predicate( HC.PREDICATE_TYPE_TAG, 'tag', inclusive = False )
self.assertEqual( p.GetUnicode(), u'-tag' )
p = HC.Predicate( HC.PREDICATE_TYPE_TAG, 'tag', inclusive = False, counts = { HC.CURRENT : 1, HC.PENDING : 2 } )
self.assertEqual( p.GetUnicode( with_count = False ), u'-tag' )
self.assertEqual( p.GetUnicode( with_count = True ), u'-tag (1) (+2)' )
#
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_AGE, ( '<', 1, 2, 3, 4 ) ) )
self.assertEqual( p.GetUnicode(), u'system:age < 1y2m3d4h' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_AGE, ( u'\u2248', 1, 2, 3, 4 ) ) )
self.assertEqual( p.GetUnicode(), u'system:age ' + u'\u2248' + ' 1y2m3d4h' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_AGE, ( '>', 1, 2, 3, 4 ) ) )
self.assertEqual( p.GetUnicode(), u'system:age > 1y2m3d4h' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_ARCHIVE, None ), counts = { HC.CURRENT : 1000 } )
self.assertEqual( p.GetUnicode(), u'system:archive (1,000)' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_DURATION, ( '<', 1000 ) ) )
self.assertEqual( p.GetUnicode(), u'system:duration < 1,000' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_EVERYTHING, None ), counts = { HC.CURRENT : 2000 } )
self.assertEqual( p.GetUnicode(), u'system:everything (2,000)' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_FILE_SERVICE, ( True, HC.CURRENT, HC.LOCAL_FILE_SERVICE_KEY ) ) )
self.assertEqual( p.GetUnicode(), u'system:is currently in local files' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_FILE_SERVICE, ( False, HC.PENDING, HC.LOCAL_FILE_SERVICE_KEY ) ) )
self.assertEqual( p.GetUnicode(), u'system:is not pending to local files' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_HASH, 'abcd'.decode( 'hex' ) ) )
self.assertEqual( p.GetUnicode(), u'system:hash is abcd' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_HEIGHT, ( '<', 2000 ) ) )
self.assertEqual( p.GetUnicode(), u'system:height < 2,000' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_INBOX, None ), counts = { HC.CURRENT : 1000 } )
self.assertEqual( p.GetUnicode(), u'system:inbox (1,000)' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_LIMIT, 2000 ) )
self.assertEqual( p.GetUnicode(), u'system:limit is 2,000' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_LOCAL, None ), counts = { HC.CURRENT : 100 } )
self.assertEqual( p.GetUnicode(), u'system:local (100)' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_MIME, HC.IMAGES ) )
self.assertEqual( p.GetUnicode(), u'system:mime is image' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_MIME, HC.VIDEO_WEBM ) )
self.assertEqual( p.GetUnicode(), u'system:mime is video/webm' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_NOT_LOCAL, None ), counts = { HC.CURRENT : 100 } )
self.assertEqual( p.GetUnicode(), u'system:not local (100)' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_NUM_TAGS, ( '<', 2 ) ) )
self.assertEqual( p.GetUnicode(), u'system:number of tags < 2' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_NUM_WORDS, ( '<', 5000 ) ) )
self.assertEqual( p.GetUnicode(), u'system:number of words < 5,000' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_RATING, ( HC.LOCAL_FILE_SERVICE_KEY, '>', 0.2 ) ) )
self.assertEqual( p.GetUnicode(), u'system:rating for local files > 0.2' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_RATIO, ( '=', 16, 9 ) ) )
self.assertEqual( p.GetUnicode(), u'system:ratio = 16:9' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_SIMILAR_TO, ( 'abcd'.decode( 'hex' ), 5 ) ) )
self.assertEqual( p.GetUnicode(), u'system:similar to abcd using max hamming of 5' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_SIZE, ( '>', 5, 1048576 ) ) )
self.assertEqual( p.GetUnicode(), u'system:size > 5MB' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_UNTAGGED, HC.IMAGES ) )
self.assertEqual( p.GetUnicode(), u'system:untagged' )
p = HC.Predicate( HC.PREDICATE_TYPE_SYSTEM, ( HC.SYSTEM_PREDICATE_TYPE_WIDTH, ( '=', 1920 ) ) )
self.assertEqual( p.GetUnicode(), u'system:width = 1,920' )
#
p = HC.Predicate( HC.PREDICATE_TYPE_NAMESPACE, 'series' )
self.assertEqual( p.GetUnicode(), u'series:*' )
p = HC.Predicate( HC.PREDICATE_TYPE_TAG, 'series', inclusive = False )
self.assertEqual( p.GetUnicode(), u'-series' )
#
p = HC.Predicate( HC.PREDICATE_TYPE_WILDCARD, 'a*i:o*' )
self.assertEqual( p.GetUnicode(), u'a*i:o*' )
p = HC.Predicate( HC.PREDICATE_TYPE_TAG, 'a*i:o*', inclusive = False )
self.assertEqual( p.GetUnicode(), u'-a*i:o*' )
#
p = HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'series:game of thrones' )
self.assertEqual( p.GetUnicode(), u' series:game of thrones' )
class TestTagParents( unittest.TestCase ):
@classmethod
def setUpClass( self ):
self._first_key = os.urandom( 32 )
self._second_key = os.urandom( 32 )
self._third_key = os.urandom( 32 )
first_dict = HC.default_dict_set()
first_dict[ HC.CURRENT ] = { ( 'current_a', 'current_b' ), ( 'child', 'mother' ), ( 'child', 'father' ), ( 'sister', 'mother' ), ( 'sister', 'father' ), ( 'brother', 'mother' ), ( 'brother', 'father' ), ( 'mother', 'grandmother' ), ( 'mother', 'grandfather' ), ( 'aunt', 'grandmother' ), ( 'aunt', 'grandfather' ), ( 'cousin', 'aunt' ), ( 'cousin', 'uncle' ), ( 'closed_loop', 'closed_loop' ), ( 'loop_a', 'loop_b' ), ( 'loop_b', 'loop_c' ) }
first_dict[ HC.DELETED ] = { ( 'deleted_a', 'deleted_b' ) }
second_dict = HC.default_dict_set()
second_dict[ HC.CURRENT ] = { ( 'loop_c', 'loop_a' ) }
second_dict[ HC.DELETED ] = { ( 'current_a', 'current_b' ) }
second_dict[ HC.PENDING ] = { ( 'pending_a', 'pending_b' ) }
second_dict[ HC.PETITIONED ] = { ( 'petitioned_a', 'petitioned_b' ) }
third_dict = HC.default_dict_set()
third_dict[ HC.CURRENT ] = { ( 'petitioned_a', 'petitioned_b' ) }
third_dict[ HC.DELETED ] = { ( 'pending_a', 'pending_b' ) }
tag_parents = collections.defaultdict( HC.default_dict_set )
tag_parents[ self._first_key ] = first_dict
tag_parents[ self._second_key ] = second_dict
tag_parents[ self._third_key ] = third_dict
HC.app.SetRead( 'tag_parents', tag_parents )
self._tag_parents_manager = HydrusTags.TagParentsManager()
def test_expand_predicates( self ):
predicates = []
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'grandmother', counts = { HC.CURRENT : 10 } ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'grandfather', counts = { HC.CURRENT : 15 } ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'not_exist', counts = { HC.CURRENT : 20 } ) )
self.assertEqual( self._tag_parents_manager.ExpandPredicates( HC.COMBINED_TAG_SERVICE_KEY, predicates ), predicates )
predicates = []
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'child', counts = { HC.CURRENT : 10 } ) )
results = []
results.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'child', counts = { HC.CURRENT : 10 } ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'mother' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'father' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'grandmother' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'grandfather' ) )
self.assertEqual( set( self._tag_parents_manager.ExpandPredicates( HC.COMBINED_TAG_SERVICE_KEY, predicates ) ), set( results ) )
predicates = []
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_NAMESPACE, 'series' ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'child', counts = { HC.CURRENT : 10 } ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'cousin', counts = { HC.CURRENT : 5 } ) )
results = []
results.append( HC.Predicate( HC.PREDICATE_TYPE_NAMESPACE, 'series' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'child', counts = { HC.CURRENT : 10 } ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'mother' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'father' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'grandmother' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'grandfather' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'cousin', counts = { HC.CURRENT : 5 } ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'aunt' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'uncle' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'grandmother' ) )
results.append( HC.Predicate( HC.PREDICATE_TYPE_PARENT, 'grandfather' ) )
self.assertEqual( set( self._tag_parents_manager.ExpandPredicates( HC.COMBINED_TAG_SERVICE_KEY, predicates ) ), set( results ) )
def test_expand_tags( self ):
tags = { 'grandmother', 'grandfather' }
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, tags ), tags )
tags = { 'child', 'cousin' }
results = { 'child', 'mother', 'father', 'grandmother', 'grandfather', 'cousin', 'aunt', 'uncle', 'grandmother', 'grandfather' }
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, tags ), results )
def test_grandparents( self ):
self.assertEqual( set( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'child' ) ), { 'mother', 'father', 'grandmother', 'grandfather' } )
self.assertEqual( set( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'mother' ) ), { 'grandmother', 'grandfather' } )
self.assertEqual( set( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'grandmother' ) ), set() )
def test_current_overwrite( self ):
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'current_a' ), [ 'current_b' ] )
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'current_b' ), [] )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'current_a' ] ), { 'current_a', 'current_b' } )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'current_b' ] ), { 'current_b' } )
def test_deleted( self ):
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'deleted_a' ), [] )
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'deleted_b' ), [] )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'deleted_a' ] ), { 'deleted_a' } )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'deleted_b' ] ), { 'deleted_b' } )
def test_no_loop( self ):
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'closed_loop' ), [] )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'closed_loop' ] ), { 'closed_loop' } )
def test_not_exist( self ):
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'not_exist' ), [] )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'not_exist' ] ), { 'not_exist' } )
def test_pending_overwrite( self ):
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'pending_a' ), [ 'pending_b' ] )
self.assertEqual( self._tag_parents_manager.GetParents( HC.COMBINED_TAG_SERVICE_KEY, 'pending_b' ), [] )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'pending_a' ] ), { 'pending_a', 'pending_b' } )
self.assertEqual( self._tag_parents_manager.ExpandTags( HC.COMBINED_TAG_SERVICE_KEY, [ 'pending_b' ] ), { 'pending_b' } )
class TestTagSiblings( unittest.TestCase ):
@classmethod
def setUpClass( self ):
self._first_key = os.urandom( 32 )
self._second_key = os.urandom( 32 )
self._third_key = os.urandom( 32 )
tag_siblings = collections.defaultdict( HC.default_dict_set )
first_dict = HC.default_dict_set()
first_dict[ HC.CURRENT ] = { ( 'ishygddt', 'i sure hope you guys don\'t do that' ), ( 'character:rei ayanami', 'character:ayanami rei' ), ( 'tree_1', 'tree_3' ), ( 'tree_2', 'tree_3' ), ( 'tree_3', 'tree_5' ), ( 'tree_4', 'tree_5' ), ( 'tree_5', 'tree_6' ), ( 'current_a', 'current_b' ), ( 'chain_a', 'chain_b' ), ( 'chain_b', 'chain_c' ), ( 'closed_loop', 'closed_loop' ), ( 'loop_a', 'loop_b' ), ( 'loop_b', 'loop_c' ) }
first_dict[ HC.DELETED ] = { ( 'deleted_a', 'deleted_b' ) }
second_dict = HC.default_dict_set()
second_dict[ HC.CURRENT ] = { ( 'loop_c', 'loop_a' ) }
second_dict[ HC.DELETED ] = { ( 'current_a', 'current_b' ) }
second_dict[ HC.PENDING ] = { ( 'pending_a', 'pending_b' ) }
second_dict[ HC.PETITIONED ] = { ( 'petitioned_a', 'petitioned_b' ) }
third_dict = HC.default_dict_set()
third_dict[ HC.CURRENT ] = { ( 'petitioned_a', 'petitioned_b' ) }
third_dict[ HC.DELETED ] = { ( 'pending_a', 'pending_b' ) }
tag_siblings[ self._first_key ] = first_dict
tag_siblings[ self._second_key ] = second_dict
tag_siblings[ self._third_key ] = third_dict
HC.app.SetRead( 'tag_siblings', tag_siblings )
self._tag_siblings_manager = HydrusTags.TagSiblingsManager()
def test_autocomplete( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'ishy' ) ), set( [ 'ishygddt', 'i sure hope you guys don\'t do that' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'i su' ) ), set( [ 'ishygddt', 'i sure hope you guys don\'t do that' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'ayan' ) ), set( [ 'character:rei ayanami', 'character:ayanami rei' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'rei' ) ), set( [ 'character:rei ayanami', 'character:ayanami rei' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'character:ayan' ) ), set( [ 'character:rei ayanami', 'character:ayanami rei' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'character:rei' ) ), set( [ 'character:rei ayanami', 'character:ayanami rei' ] ) )
def test_collapse_namespace( self ):
self.assertEqual( self._tag_siblings_manager.CollapseNamespacedTags( 'character', [ 'ayanami rei', 'rei ayanami', 'ikari shinji' ] ), set( [ 'ayanami rei', 'ikari shinji' ] ) )
def test_collapse_predicates( self ):
predicates = []
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'chain_a', counts = { HC.CURRENT : 10 } ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'chain_b', counts = { HC.CURRENT : 5 } ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'chain_c', counts = { HC.CURRENT : 20 } ) )
results = [ HC.Predicate( HC.PREDICATE_TYPE_TAG, 'chain_c', counts = { HC.CURRENT : 35 } ) ]
self.assertEqual( self._tag_siblings_manager.CollapsePredicates( predicates ), results )
predicates = []
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'chain_a', counts = { HC.CURRENT : 10 } ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'chain_b', counts = { HC.CURRENT : 5 } ) )
predicates.append( HC.Predicate( HC.PREDICATE_TYPE_TAG, 'chain_c', counts = { HC.CURRENT : 20 } ) )
( result, ) = self._tag_siblings_manager.CollapsePredicates( predicates )
self.assertEqual( result.GetCount(), 35 )
def test_chain( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'chai' ) ), set( [ 'chain_a', 'chain_b', 'chain_c' ] ) )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'chain_a' ), 'chain_c' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'chain_b' ), 'chain_c' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'chain_c' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'chain_a' ) ), set( [ 'chain_a', 'chain_b', 'chain_c' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'chain_b' ) ), set( [ 'chain_a', 'chain_b', 'chain_c' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'chain_c' ) ), set( [ 'chain_a', 'chain_b', 'chain_c' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'chain_a', 'chain_b' ] ) ), set( [ 'chain_c' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'chain_a', 'chain_b', 'chain_c' ] ) ), set( [ 'chain_c' ] ) )
# collapsetagstocount
def test_current_overwrite( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'curr' ) ), set( [ 'current_a', 'current_b' ] ) )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'current_a' ), 'current_b' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'current_b' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'current_a' ) ), set( [ 'current_a', 'current_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'current_b' ) ), set( [ 'current_a', 'current_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'chain_a', 'chain_b' ] ) ), set( [ 'chain_c' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'chain_a', 'chain_b', 'chain_c' ] ) ), set( [ 'chain_c' ] ) )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'chain_a' : 10, 'chain_b' : 5 } ), { 'chain_c' : 15 } )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'chain_a' : 10, 'chain_b' : 5, 'chain_c' : 20 } ), { 'chain_c' : 35 } )
def test_deleted( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'dele' ) ), set() )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'deleted_a' ), None )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'deleted_b' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'deleted_a' ) ), set( [ 'deleted_a' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'deleted_b' ) ), set( [ 'deleted_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'deleted_a', 'deleted_b' ] ) ), set( [ 'deleted_a', 'deleted_b' ] ) )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'deleted_a' : 10, 'deleted_b' : 5 } ), { 'deleted_a' : 10, 'deleted_b' : 5 } )
def test_no_loop( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'loop' ) ), set( [ 'loop_a', 'loop_b', 'loop_c' ] ) )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'closed_loop' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'closed_loop' ) ), set( [ 'closed_loop' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'closed_loop' ] ) ), set( [ 'closed_loop' ] ) )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'closed_loop' : 10 } ), { 'closed_loop' : 10 } )
def test_not_exist( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'not_' ) ), set() )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'not_exist' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'not_exist' ) ), set( [ 'not_exist' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'not_exist' ] ) ), set( [ 'not_exist' ] ) )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'not_exist' : 10 } ), { 'not_exist' : 10 } )
def test_pending_overwrite( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'pend' ) ), set( [ 'pending_a', 'pending_b' ] ) )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'pending_a' ), 'pending_b' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'pending_b' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'pending_a' ) ), set( [ 'pending_a', 'pending_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'pending_b' ) ), set( [ 'pending_a', 'pending_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'pending_a' ] ) ), set( [ 'pending_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'pending_a', 'pending_b' ] ) ), set( [ 'pending_b' ] ) )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'pending_a' : 10 } ), { 'pending_b' : 10 } )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'pending_a' : 10, 'pending_b' : 5 } ), { 'pending_b' : 15 } )
def test_petitioned_no_overwrite( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'petitioned_a' ) ), set( [ 'petitioned_a', 'petitioned_b' ] ) )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'petitioned_a' ), 'petitioned_b' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'petitioned_b' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'petitioned_a' ) ), set( [ 'petitioned_a', 'petitioned_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'petitioned_b' ) ), set( [ 'petitioned_a', 'petitioned_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'petitioned_a' ] ) ), set( [ 'petitioned_b' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'petitioned_a', 'petitioned_b' ] ) ), set( [ 'petitioned_b' ] ) )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'petitioned_a' : 10 } ), { 'petitioned_b' : 10 } )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'petitioned_a' : 10, 'petitioned_b' : 5 } ), { 'petitioned_b' : 15 } )
def test_tree( self ):
self.assertEqual( set( self._tag_siblings_manager.GetAutocompleteSiblings( 'tree' ) ), set( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'tree_1' ), 'tree_6' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'tree_2' ), 'tree_6' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'tree_3' ), 'tree_6' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'tree_4' ), 'tree_6' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'tree_5' ), 'tree_6' )
self.assertEqual( self._tag_siblings_manager.GetSibling( 'tree_6' ), None )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'tree_1' ) ), set( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'tree_2' ) ), set( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'tree_3' ) ), set( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'tree_4' ) ), set( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'tree_5' ) ), set( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.GetAllSiblings( 'tree_6' ) ), set( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'tree_1' ] ) ), set( [ 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'tree_1', 'tree_3', 'tree_5' ] ) ), set( [ 'tree_6' ] ) )
self.assertEqual( set( self._tag_siblings_manager.CollapseTags( [ 'tree_1', 'tree_2', 'tree_3', 'tree_4', 'tree_5', 'tree_6' ] ) ), set( [ 'tree_6' ] ) )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'tree_1' : 10 } ), { 'tree_6' : 10 } )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'tree_1' : 10, 'tree_3' : 5, 'tree_5' : 20 } ), { 'tree_6' : 35 } )
self.assertEqual( self._tag_siblings_manager.CollapseTagsToCount( { 'tree_1' : 10, 'tree_2' : 3, 'tree_3' : 5, 'tree_4' : 2, 'tree_5' : 20, 'tree_6' : 30 } ), { 'tree_6' : 70 } )