from django.test import TestCase
from rest_framework.test import APIClient
from .models import User

class PredictionTest(TestCase):
    def setUp(self):
        self.client = APIClient()
        self.user = User.objects.create()