com.fiveamsolutions.nci.commons.kmplot.KMPlotServiceTest.java Source code

Java tutorial

Introduction

Here is the source code for com.fiveamsolutions.nci.commons.kmplot.KMPlotServiceTest.java

Source

/**
 * The software subject to this notice and license includes both human readable
 * source code form and machine readable, binary, object code form. The caIntegrator2
 * Software was developed in conjunction with the National Cancer Institute
 * (NCI) by NCI employees, 5AM Solutions, Inc. (5AM), ScenPro, Inc. (ScenPro)
 * and Science Applications International Corporation (SAIC). To the extent
 * government employees are authors, any rights in such works shall be subject
 * to Title 17 of the United States Code, section 105.
 *
 * This caIntegrator2 Software License (the License) is between NCI and You. You (or
 * Your) shall mean a person or an entity, and all other entities that control,
 * are controlled by, or are under common control with the entity. Control for
 * purposes of this definition means (i) the direct or indirect power to cause
 * the direction or management of such entity, whether by contract or otherwise,
 * or (ii) ownership of fifty percent (50%) or more of the outstanding shares,
 * or (iii) beneficial ownership of such entity.
 *
 * This License is granted provided that You agree to the conditions described
 * below. NCI grants You a non-exclusive, worldwide, perpetual, fully-paid-up,
 * no-charge, irrevocable, transferable and royalty-free right and license in
 * its rights in the caIntegrator2 Software to (i) use, install, access, operate,
 * execute, copy, modify, translate, market, publicly display, publicly perform,
 * and prepare derivative works of the caIntegrator2 Software; (ii) distribute and
 * have distributed to and by third parties the caIntegrator2 Software and any
 * modifications and derivative works thereof; and (iii) sublicense the
 * foregoing rights set out in (i) and (ii) to third parties, including the
 * right to license such rights to further third parties. For sake of clarity,
 * and not by way of limitation, NCI shall have no right of accounting or right
 * of payment from You or Your sub-licensees for the rights granted under this
 * License. This License is granted at no charge to You.
 *
 * Your redistributions of the source code for the Software must retain the
 * above copyright notice, this list of conditions and the disclaimer and
 * limitation of liability of Article 6, below. Your redistributions in object
 * code form must reproduce the above copyright notice, this list of conditions
 * and the disclaimer of Article 6 in the documentation and/or other materials
 * provided with the distribution, if any.
 *
 * Your end-user documentation included with the redistribution, if any, must
 * include the following acknowledgment: This product includes software
 * developed by 5AM, ScenPro, SAIC and the National Cancer Institute. If You do
 * not include such end-user documentation, You shall include this acknowledgment
 * in the Software itself, wherever such third-party acknowledgments normally
 * appear.
 *
 * You may not use the names "The National Cancer Institute", "NCI", "ScenPro",
 * "SAIC" or "5AM" to endorse or promote products derived from this Software.
 * This License does not authorize You to use any trademarks, service marks,
 * trade names, logos or product names of either NCI, ScenPro, SAID or 5AM,
 * except as required to comply with the terms of this License.
 *
 * For sake of clarity, and not by way of limitation, You may incorporate this
 * Software into Your proprietary programs and into any third party proprietary
 * programs. However, if You incorporate the Software into third party
 * proprietary programs, You agree that You are solely responsible for obtaining
 * any permission from such third parties required to incorporate the Software
 * into such third party proprietary programs and for informing Your a
 * sub-licensees, including without limitation Your end-users, of their
 * obligation to secure any required permissions from such third parties before
 * incorporating the Software into such third party proprietary software
 * programs. In the event that You fail to obtain such permissions, You agree
 * to indemnify NCI for any claims against NCI by such third parties, except to
 * the extent prohibited by law, resulting from Your failure to obtain such
 * permissions.
 *
 * For sake of clarity, and not by way of limitation, You may add Your own
 * copyright statement to Your modifications and to the derivative works, and
 * You may provide additional or different license terms and conditions in Your
 * sublicenses of modifications of the Software, or any derivative works of the
 * Software as a whole, provided Your use, reproduction, and distribution of the
 * Work otherwise complies with the conditions stated in this License.
 *
 * THIS SOFTWARE IS PROVIDED "AS IS," AND ANY EXPRESSED OR IMPLIED WARRANTIES,
 * (INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY,
 * NON-INFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE) ARE DISCLAIMED. IN NO
 * EVENT SHALL THE NATIONAL CANCER INSTITUTE, 5AM SOLUTIONS, INC., SCENPRO, INC.,
 * SCIENCE APPLICATIONS INTERNATIONAL CORPORATION OR THEIR
 * AFFILIATES BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package com.fiveamsolutions.nci.commons.kmplot;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import java.awt.Color;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.junit.Test;
import org.springframework.mock.web.MockHttpServletResponse;

public class KMPlotServiceTest {

    @Test
    public void testGeneratePlot() throws Exception {
        KMPlot plot = getPlot();
        MockHttpServletResponse response = new MockHttpServletResponse();
        plot.writePlotImage(response.getOutputStream());
        assertNotNull(response.getContentAsString());
        assertTrue(StringUtils.isNotBlank(response.getContentAsString()));
    }

    private KMPlot getPlot() {
        KMPlotConfiguration configuration = new KMPlotConfiguration();
        configuration.setTitle("title");
        configuration.setDurationLabel("duration");
        configuration.setProbabilityLabel("probability");
        configuration.getGroups().add(createGroup("group 1", 2, Color.RED));
        configuration.getGroups().add(createGroup("group 2", 5, Color.GREEN));
        configuration.getGroups().add(createGroup("group 3", 2, Color.BLUE));
        configuration.getGroups().add(createGroup("group 4", 1, Color.BLACK));
        configuration.getGroups().add(createGroup("group 5", 1, Color.GRAY));
        KMPlot plot = new KMPlotServiceImpl().generatePlot(configuration);
        assertNotNull(plot);
        assertNotNull(plot.getConfiguration());
        assertNotNull(plot.getConfiguration().getGroups());
        assertEquals(5, plot.getConfiguration().getGroups().size());
        verifyGroup(plot, 0, "group 1", 2, Color.RED);
        verifyGroup(plot, 1, "group 2", 5, Color.GREEN);
        verifyGroup(plot, 2, "group 3", 2, Color.BLUE);
        verifyGroup(plot, 3, "group 4", 1, Color.BLACK);
        verifyGroup(plot, 4, "group 5", 1, Color.GRAY);
        return plot;
    }

    private SubjectGroup createGroup(String groupName, int subjectCount, Color color) {
        SubjectGroup group = new SubjectGroup();
        group.setColor(color);
        group.setName(groupName);
        for (int i = 1; i <= subjectCount; i++) {
            SubjectSurvivalData survivalData = new SubjectSurvivalData(i, false);
            group.getSurvivalData().add(survivalData);
        }
        return group;
    }

    private void verifyGroup(KMPlot plot, int groupIndex, String groupName, int subjectCount, Color color) {
        SubjectGroup sg = plot.getConfiguration().getGroups().get(groupIndex);
        assertNotNull(sg);
        assertEquals(groupName, sg.getName());
        assertEquals(color, sg.getColor());
        assertNotNull(sg.getSurvivalData());
        assertEquals(subjectCount, sg.getSurvivalData().size());
        for (int i = 0; i < subjectCount; i++) {
            SubjectSurvivalData ssd = sg.getSurvivalData().get(i);
            assertNotNull(ssd);
            assertFalse(ssd.isCensor());
            assertEquals(i + 1, ssd.getSurvivalLength());
        }
    }

    @Test
    public void testGetAllStringPValues() {
        KMPlot plot = getPlot();
        Map<String, Map<String, String>> pValues = KMPlotUtils.retrieveAllStringPValues(plot);
        assertNull(pValues.get("group 1").get("group 1"));
        assertEquals("0.18", pValues.get("group 1").get("group 2"));
        assertEquals("1.00", pValues.get("group 1").get("group 3"));
        assertEquals("0.48", pValues.get("group 1").get("group 4"));
        assertEquals("0.48", pValues.get("group 1").get("group 5"));
        assertNull(pValues.get("group 2").get("group 1"));
        assertNull(pValues.get("group 2").get("group 2"));
        assertEquals("0.18", pValues.get("group 2").get("group 3"));
        assertEquals("0.16", pValues.get("group 2").get("group 4"));
        assertEquals("0.16", pValues.get("group 2").get("group 5"));
        assertNull(pValues.get("group 3").get("group 1"));
        assertNull(pValues.get("group 3").get("group 2"));
        assertNull(pValues.get("group 3").get("group 3"));
        assertEquals("0.48", pValues.get("group 3").get("group 4"));
        assertEquals("0.48", pValues.get("group 3").get("group 5"));
        assertNull(pValues.get("group 4").get("group 1"));
        assertNull(pValues.get("group 4").get("group 2"));
        assertNull(pValues.get("group 4").get("group 3"));
        assertNull(pValues.get("group 4").get("group 4"));
        assertEquals("N/A", pValues.get("group 4").get("group 5"));
        assertNull(pValues.get("group 5").get("group 1"));
        assertNull(pValues.get("group 5").get("group 2"));
        assertNull(pValues.get("group 5").get("group 3"));
        assertNull(pValues.get("group 5").get("group 4"));
        assertNull(pValues.get("group 5").get("group 5"));
    }
}