dwhswenson/contact_map

View on GitHub
contact_map/tests/utils.py

Summary

Maintainability
A
0 mins
Test Coverage
import os
from pkg_resources import resource_filename

# pylint: disable=unused-import

import numpy as np
from numpy.testing import assert_array_equal, assert_allclose
import pytest

def find_testfile(fname):
    return resource_filename('contact_map', os.path.join('tests', fname))

def zero_to_nan(input_arr):
    arr = input_arr.copy()
    arr[arr == 0.0] = np.nan
    return arr