def test_get_cluster_name(self, random_mock):
    cluster = ClusterConsensus()

    cluster.output_file = StringIO()
    random_mock.return_value = 10000
    self.assertEqual(cluster._get_cluster_name(), 'cluster_10000')
    
    cluster.output_file = MagicMock()
    cluster.output_file.name = 'foo/bar.baz.mfa'
    self.assertEqual(cluster._get_cluster_name(), 'bar.baz')

    cluster.cluster_name = 'foo'
    self.assertEqual(cluster._get_cluster_name(), 'foo')
Exemple #2
0
    def test_get_cluster_name(self, random_mock):
        cluster = ClusterConsensus()

        cluster.output_file = StringIO()
        random_mock.return_value = 10000
        self.assertEqual(cluster._get_cluster_name(), 'cluster_10000')

        cluster.output_file = MagicMock()
        cluster.output_file.name = 'foo/bar.baz.mfa'
        self.assertEqual(cluster._get_cluster_name(), 'bar.baz')

        cluster.cluster_name = 'foo'
        self.assertEqual(cluster._get_cluster_name(), 'foo')
  def test_write_consensus(self):
    cluster = ClusterConsensus()

    cluster.output_file = StringIO()
    cluster.cluster_name = 'cluster_1000'
    cluster._calculate_consensus = MagicMock(return_value='NNANTGUNCN')

    cluster.write_consensus()

    expected_results = """\
>cluster_1000
NNANTGUNCN
"""
    cluster.output_file.seek(0)
    self.assertEqual(cluster.output_file.read(), expected_results)
Exemple #4
0
    def test_write_consensus(self):
        cluster = ClusterConsensus()

        cluster.output_file = StringIO()
        cluster.cluster_name = 'cluster_1000'
        cluster._calculate_consensus = MagicMock(return_value='NNANTGUNCN')

        cluster.write_consensus()

        expected_results = """\
>cluster_1000
NNANTGUNCN
"""
        cluster.output_file.seek(0)
        self.assertEqual(cluster.output_file.read(), expected_results)