Example #1
0
    def test_get_counts(self):
        """get_counts should gets all the counts for an input file"""

        # test the pie charts
        img_data = get_counts("Phylum", ['14FC041', '14FC042', '14FC043'], 5,
                              "/tmp/qiimewebfiles/", 1, self.lines_parsed,
                              self.prefs, self.color_prefs, 'black', 'white',
                              'pie', self.generate_image_type, self.plot_width,
                              self.plot_height, self.bar_width, self.dpi,
                              'file.txt', 0, 'categorical', False, False)

        self.assertEqual(len(img_data), 8)

        # test the area charts
        img_data = get_counts("Phylum", ['14FC041', '14FC042', '14FC043'], 5,
                              "/tmp/qiimewebfiles/", 1, self.lines_parsed,
                              self.prefs, self.color_prefs, 'black', 'white',
                              'area', self.generate_image_type,
                              self.plot_width, self.plot_height,
                              self.bar_width, self.dpi, 'file.txt', 0,
                              'categorical', False, False)

        self.assertEqual(len(img_data), 2)

        # test the area charts
        img_data = get_counts("Phylum", ['14FC041', '14FC042', '14FC043'], 5,
                              "/tmp/qiimewebfiles/", 1, self.lines_parsed,
                              self.prefs, self.color_prefs, 'black', 'white',
                              'bar', self.generate_image_type, self.plot_width,
                              self.plot_height, self.bar_width, self.dpi,
                              'file.txt', 0, 'categorical', False, False)

        self.assertEqual(len(img_data), 2)

        # clean up files generated
        self._paths_to_clean_up = [
            "/tmp/qiimewebfiles/charts/" + f
            for f in listdir("/tmp/qiimewebfiles/charts")
        ]
    def test_get_counts(self):
        """get_counts should gets all the counts for an input file"""
        
        #test the pie charts
        img_data = get_counts("Phylum",['14FC041','14FC042','14FC043'],5,\
                              "/tmp/qiimewebfiles/",1,self.lines_parsed,\
                              self.prefs,self.color_prefs,'black','white',\
                              'pie',self.generate_image_type,self.plot_width,\
                              self.plot_height,self.bar_width,self.dpi,\
                              'file.txt',0,'categorical',False,False)
    
        self.assertEqual(len(img_data), 8)
        
        #test the area charts
        img_data = get_counts("Phylum",['14FC041','14FC042','14FC043'],5,\
                              "/tmp/qiimewebfiles/",1,self.lines_parsed,\
                              self.prefs,self.color_prefs,'black','white',\
                              'area',self.generate_image_type,self.plot_width,\
                              self.plot_height,self.bar_width,self.dpi,\
                              'file.txt',0,'categorical',False,False)

        self.assertEqual(len(img_data), 2)
        
        #test the area charts
        img_data = get_counts("Phylum",['14FC041','14FC042','14FC043'],5,\
                              "/tmp/qiimewebfiles/",1,self.lines_parsed,\
                              self.prefs,self.color_prefs,'black','white',\
                              'bar',self.generate_image_type,self.plot_width,\
                              self.plot_height,self.bar_width,self.dpi,\
                              'file.txt',0,'categorical',False,False)

        self.assertEqual(len(img_data), 2)
         
        #clean up files generated
        self._paths_to_clean_up = ["/tmp/qiimewebfiles/charts/"+f \
                                  for f in listdir("/tmp/qiimewebfiles/charts")] 
    def test_get_counts(self):
        """get_counts should gets all the counts for an input file"""

        # test the pie charts
        img_data = get_counts(
            "Phylum",
            ["14FC041", "14FC042", "14FC043"],
            5,
            "/tmp/qiimewebfiles/",
            1,
            self.lines_parsed,
            self.prefs,
            self.color_prefs,
            "black",
            "white",
            "pie",
            self.generate_image_type,
            self.plot_width,
            self.plot_height,
            self.bar_width,
            self.dpi,
            "file.txt",
            0,
            "categorical",
            False,
        )

        self.assertEqual(len(img_data), 8)

        # test the area charts
        img_data = get_counts(
            "Phylum",
            ["14FC041", "14FC042", "14FC043"],
            5,
            "/tmp/qiimewebfiles/",
            1,
            self.lines_parsed,
            self.prefs,
            self.color_prefs,
            "black",
            "white",
            "area",
            self.generate_image_type,
            self.plot_width,
            self.plot_height,
            self.bar_width,
            self.dpi,
            "file.txt",
            0,
            "categorical",
            False,
        )

        self.assertEqual(len(img_data), 2)

        # test the area charts
        img_data = get_counts(
            "Phylum",
            ["14FC041", "14FC042", "14FC043"],
            5,
            "/tmp/qiimewebfiles/",
            1,
            self.lines_parsed,
            self.prefs,
            self.color_prefs,
            "black",
            "white",
            "bar",
            self.generate_image_type,
            self.plot_width,
            self.plot_height,
            self.bar_width,
            self.dpi,
            "file.txt",
            0,
            "categorical",
            False,
        )

        self.assertEqual(len(img_data), 2)

        # clean up files generated
        self._paths_to_clean_up = ["/tmp/qiimewebfiles/charts/" + f for f in listdir("/tmp/qiimewebfiles/charts")]