blob: 42aa90923341294c116cc1c9cd1eac109cb6c5d8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
"""Tests wqflask/base/species.py"""
import unittest
from unittest import mock
from base.species import TheSpecies
from base.species import IndChromosome
class TestTheSpecies(unittest.TestCase):
"""Tests for TheSpecies class"""
@mock.patch('base.species.Chromosomes')
def test_create_species_with_null_species_name(self, mock_chromosome):
"""Test that TheSpecies is instantiated correctly when the
species_name is provided."""
mock_chromosome.return_value = 1
test_species = TheSpecies(dataset="random_dataset", species_name="a")
self.assertEqual(test_species.name, "a")
self.assertEqual(test_species.chromosomes, 1)
@mock.patch('base.species.Chromosomes')
def test_create_species_with_species_name(self, mock_chromosome):
"""Test that TheSpecies is instantiated correctly when the
species_name is not provided."""
mock_chromosome.return_value = 1
test_species = TheSpecies(dataset="random_dataset")
self.assertEqual(test_species.dataset, "random_dataset")
self.assertEqual(test_species.chromosomes, 1)
mock_chromosome.assert_called_once_with(dataset="random_dataset")
class TestIndChromosome(unittest.TestCase):
"""Tests for IndChromosome class"""
def test_create_ind_chromosome(self):
"""Test that IndChromosome is instantiated correctly"""
test_ind_chromosome = IndChromosome(name="Test", length=10000000)
self.assertEqual(test_ind_chromosome.name, "Test")
self.assertEqual(test_ind_chromosome.length, 10000000)
self.assertEqual(test_ind_chromosome.mb_length, 10)
|
