stroopTest.py

import pygame
import random
import time

pygame.init()

# Set up display
screen = pygame.display.set_mode((600, 400))
pygame.display.set_caption("Stroop Test")

# Define colors
colors = {'RED': (255, 0, 0), 'GREEN': (0, 255, 0), 'BLUE': (0, 0, 255), 'YELLOW': (255, 255, 0)}
color_names = list(colors.keys())

font = pygame.font.SysFont(None, 80)

def draw_text(word, color):
    screen.fill((255, 255, 255))
    text = font.render(word, True, color)
    rect = text.get_rect(center=(300, 200))
    screen.blit(text, rect)
    pygame.display.flip()

def stroop_trial():
    word = random.choice(color_names)
    ink = colors[random.choice(color_names)]
    correct_color = [k for k, v in colors.items() if v == ink][0]

    draw_text(word, ink)

    start_time = time.time()
    waiting = True
    user_input = None

    while waiting:
        for event in pygame.event.get():
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_r:
                    user_input = 'RED'
                    waiting = False
                elif event.key == pygame.K_g:
                    user_input = 'GREEN'
                    waiting = False
                elif event.key == pygame.K_b:
                    user_input = 'BLUE'
                    waiting = False
                elif event.key == pygame.K_y:
                    user_input = 'YELLOW'
                    waiting = False
            elif event.type == pygame.QUIT:
                pygame.quit()
                exit()

    reaction_time = time.time() - start_time
    correct = (user_input == correct_color)
    return reaction_time, correct

# Run 5 trials
results = []
for i in range(5):
    pygame.time.delay(1000)
    rt, correct = stroop_trial()
    results.append((rt, correct))

pygame.quit()

# Print results
for i, (rt, correct) in enumerate(results):
    print(f"Trial {i+1}: {'Correct' if correct else 'Wrong'} – Reaction Time: {rt:.2f} seconds")