import sys,unittest,threading
from pathlib import Path
sys.path.insert(0,str(Path(__file__).resolve().parents[1]/'engine'))
import generation,geometry
class Generation(unittest.TestCase):
 def test_preserves_controls_and_complete_prefix(self):
  g=geometry.new();seen=[]
  def gen(adjusted,attempt):
   seen.append(adjusted['controls'].copy())
   return {'text':'The machine feared the void. '+('The cold blood '*12)}
  r=generation.rewrite('The machine predicts sensory information.',gen,g,True,threading.Event(),lambda s:None)
  self.assertEqual(seen,[g['controls']]);self.assertEqual(r['text'],'The machine feared the void.')
  self.assertEqual(len(r['attempts']),1);self.assertIn('cold blood',r['attempts'][0]['text'])
 def test_shared_allowance_is_used(self):
  r=generation.rewrite('Short.',lambda *a:{'text':'A much longer complete sentence.'},geometry.new(),True,threading.Event(),lambda s:None,max_chars=100)
  self.assertEqual(r['text'],'A much longer complete sentence.');self.assertNotIn('original_retained',r)
 def test_no_complete_sentence_is_flagged(self):
  r=generation.rewrite('Original.',lambda *a:{'text':'cut off'},geometry.new(),True,threading.Event(),lambda s:None)
  self.assertTrue(r['original_retained']);self.assertEqual(len(r['attempts']),1)
 def test_repetition_detection(self):
  self.assertTrue(generation.repetition('cold blood of the machine '*8))
  self.assertFalse(generation.repetition('The machine watches the rain. It hopes for quiet.'))

 def test_concise_paragraph_allowance(self):
  import documents
  text='A'*600+' '+'B'*150
  spans=[(0,600),(601,751)]
  first=documents.rewrite_allowance(text,spans,{},0,multiplier=1.25)
  self.assertEqual(first,750)
  records={'0':{'text':'X'*650}}
  last=documents.rewrite_allowance(text,spans,records,1,multiplier=1.25)
  records['1']={'text':'Y'*last}
  self.assertLessEqual(len(documents.assemble(text,spans,records)),int(1.25*len(text)))