def test_in(self):
self.assertListEqual(
methodfinder._find([[1, 2, 6, 7], 6], expected_value=True),
[
'6 in [1, 2, 6, 7]\n# to use the "in" syntax with your own type, override __contains__'
],
)
def test_zip2(self):
self.assertListEqual(
methodfinder._find(
[itertools, [1, 2], [3, 4]], expected_value=[[1, 3], [2, 4]]
),
["itertools.zip_longest([1, 2], [3, 4])"],
)
def test_numpy(self):
import numpy as np
self.assertListEqual(
methodfinder._find(
[3, np],
expected_value=np.array(
[[1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 1.0]]
),
),
["numpy.eye(3)", "numpy.identity(3)"],
)
def test_one_one_one(self):
self.assertListEqual(
methodfinder._find([1, 1], expected_value=1),
[
"1&1",
"1**1",
"1*1",
"1.denominator",
"1.numerator",
"1.real",
"1//1",
"1|1",
"max(1, 1)",
"min(1, 1)",
"pow(1, 1)",
"round(1, 1)",
],
)
def test_add2(self):
self.assertListEqual(
methodfinder._find([1, 2], expected_value=3),
["1+2", "1^2", "1|2", "2+1", "2^1", "2|1"],
)
def test_doublenegative(self):
self.assertListEqual(
methodfinder._find([-1], expected_value=1),
["-(-1)", "-1.bit_length()", "-1.denominator", "abs(-1)"],
)
def test_divide(self):
self.assertListEqual(methodfinder._find([3, 2], expected_value=1.5), ["3/2"])
def test_add(self):
self.assertListEqual(
methodfinder._find([-1, 3], expected_value=2), ["-1%3", "-1+3", "3+-1"]
)
def test_str(self):
self.assertListEqual(
methodfinder._find([3], expected_value="3"),
["ascii(3)", "format(3)", "repr(3)", "str(3)"],
)
def test_any(self):
self.assertListEqual(
methodfinder._find([[]], expected_value=False),
["any([])", "bool([])", "callable([])"],
)
def test_len(self):
self.assertListEqual(
methodfinder._find([[]], expected_value=0), ["len([])", "sum([])"]
)
def test_zip(self):
self.assertListEqual(
methodfinder._find([[1, 2], [3, 4]], expected_value=[[1, 3], [2, 4]]),
["zip([1, 2], [3, 4])"],
)
def test_permutations(self):
self.assertListEqual(
methodfinder._find([itertools, [1, 2]], expected_value=[[1, 2], [2, 1]]),
["itertools.permutations([1, 2])"],
)
def test_join(self):
self.assertListEqual(
methodfinder._find([" ", ["foo", "bar"]], expected_value="foo bar"),
["' '.join(['foo', 'bar'])"],
)
def test_hasattr(self):
self.assertListEqual(
methodfinder._find([[1, 2], "__iter__"], expected_value=True),
["hasattr([1, 2], '__iter__')"],
)
