def test__rebuild_map(test_data): """ Test private _rebuild_map method. """ for d in test_data: gpm = GenotypePhenotypeMap(wildtype=d["wildtype"], genotype=d["genotype"]) # Trash all the stuff that _rebuild_map builds from scratch gpm._encoding_table = None gpm._data = gpm.data.drop(["binary"], axis=1) gpm._mutant = None gpm._data = gpm.data.drop(["n_mutations"], axis=1) gpm._data = gpm.data.drop(["name"], axis=1) gpm._rebuild_map() # Lazy check of encoding table. The encoding table is a pretty # complicated object. See test_utils.test_get_encoding_table for test # of its creation. assert type(gpm._encoding_table) is pd.DataFrame assert np.array_equal(d["binary"], gpm.binary) assert gpm.mutant == d["mutant"] assert np.array_equal(d["n_mutations"], gpm.n_mutations) assert np.array_equal(d["name"], gpm.name)
def test__add_mutant(test_data): """ Test private _add_mutant method. """ for d in test_data: gpm = GenotypePhenotypeMap(wildtype=d["wildtype"], genotype=d["genotype"]) # Get rid of the mutant gpm._mutant = None gpm._add_mutant() assert np.array_equal(gpm.mutant, d["mutant"])