def test_gen_mac(self): with patch("random.randint", return_value=1) as random_mock: self.assertEqual(random_mock.return_value, 1) ret = network.gen_mac("00:16:3E") expected_mac = "00:16:3E:01:01:01" self.assertEqual(ret, expected_mac)
def test_gen_mac(self): with patch('random.randint', return_value=1) as random_mock: self.assertEqual(random_mock.return_value, 1) ret = network.gen_mac('00:16:3E') expected_mac = '00:16:3E:01:01:01' self.assertEqual(ret, expected_mac)