Skip to content

Comments

Add comprehensive test suite for rewards, detection, and utilities#344

Open
amussara wants to merge 1 commit intoBitMind-AI:mainfrom
amussara:add-comprehensive-tests
Open

Add comprehensive test suite for rewards, detection, and utilities#344
amussara wants to merge 1 commit intoBitMind-AI:mainfrom
amussara:add-comprehensive-tests

Conversation

@amussara
Copy link

5 new test files covering rewards, duplicate detection, types, cache, and utils. 1,098 lines.

…and duplicate detection

- tests/test_rewards.py: 25+ tests covering discriminator rewards, generator
  base rewards, generator reward multipliers, liveness filtering, edge cases
  (empty/None inputs, invalid types, negative MCC clamping, custom weights)
- tests/test_duplicate_detection.py: Tests for hash extraction, hamming distance,
  duplicate detection logic, crop-resistant matching, find_duplicates
- tests/test_types.py: Tests for Modality, MediaType, SourceType enums,
  DatasetConfig validation, ModelConfig construction, DiscriminatorModelId
  serialization roundtrip
- tests/test_cache_types.py: Tests for PromptEntry, MediaEntry, Media,
  VerificationResult dataclasses and their serialization
- tests/test_utils.py: Tests for get_file_modality, get_metadata, ExitContext,
  fail_with_none decorator, run_in_thread
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant