test_gene.py

from django.contrib.auth.models import User
from django.test import TestCase
from django.urls import reverse

import pandas as pd
from rest_framework import status
from rest_framework_jwt.settings import api_settings

from metagenedb.api.catalog.views.gene import GeneViewSet


class TestGenes(TestCase):

    def test_get_genes_no_auth(self):
        """
        Unauthenticated users should be able to access genes
        @TODO make unaccessible
        """
        url = reverse('api:catalog:v1:genes-list')
        resp = self.client.get(url)
        self.assertEqual(resp.status_code, status.HTTP_200_OK)

    def test_get_genes_auth(self):
        """
        Authenticated users should be able to access genes
        """

        jwt_payload_handler = api_settings.JWT_PAYLOAD_HANDLER
        jwt_encode_handler = api_settings.JWT_ENCODE_HANDLER

        user = User.objects.create_user(username='user', email='user@foo.com', password='pass')
        user.is_active = True
        user.save()
        payload = jwt_payload_handler(user)
        token = jwt_encode_handler(payload)

        url = reverse('api:catalog:v1:genes-list')
        resp = self.client.get(url, format='json', HTTP_AUTHORIZATION=f"JWT {token}")
        self.assertEqual(resp.status_code, status.HTTP_200_OK)


class TestCountWindows(TestCase):

    def setUp(self):
        self.window_col = "length"
        self.df = pd.DataFrame(
            [22, 29, 35],
            columns=[self.window_col]
        )

    def test_simple_count_window10(self):
        expected_dict = {
            'labels': ['<0.01k', '0.01k-0.02k', '0.02k-0.03k', '>0.03k'],
            'counts': [0, 0, 2, 1]
        }
        geneviewset = GeneViewSet()
        test_dict = geneviewset._count_windows(self.df, 10, window_col=self.window_col)
        self.assertDictEqual(test_dict, expected_dict)

    def test_simple_count_window10_stop20(self):
        expected_dict = {
            'labels': ['<0.01k', '0.01k-0.02k', '>0.02k'],
            'counts': [0, 0, 3]
        }
        geneviewset = GeneViewSet()
        test_dict = geneviewset._count_windows(self.df, window_size=10,
                                               window_col=self.window_col, stop_at=20)
        self.assertDictEqual(test_dict, expected_dict)