Skip to content

Commit aaa79a5

Browse files
committed
Merge pull request #72 from ddemidov/issue-70
Move platform-guessing from __init__.py to auto.py
2 parents 31efa43 + 8c654d6 commit aaa79a5

File tree

4 files changed

+30
-30
lines changed

4 files changed

+30
-30
lines changed

ev3dev/__init__.py

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +0,0 @@
1-
import platform
2-
3-
# -----------------------------------------------------------------------------
4-
# Guess platform we are running on
5-
def current_platform():
6-
machine = platform.machine()
7-
if machine == 'armv5tejl':
8-
return 'ev3'
9-
elif machine == 'armv6l':
10-
return 'brickpi'
11-
else:
12-
return 'unsupported'
13-
14-
if current_platform() == 'brickpi':
15-
from .brickpi import *
16-
else:
17-
# Import ev3 by default, so that it is covered by documentation.
18-
from .ev3 import *

ev3dev/auto.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
import platform
2+
3+
# -----------------------------------------------------------------------------
4+
# Guess platform we are running on
5+
def current_platform():
6+
machine = platform.machine()
7+
if machine == 'armv5tejl':
8+
return 'ev3'
9+
elif machine == 'armv6l':
10+
return 'brickpi'
11+
else:
12+
return 'unsupported'
13+
14+
if current_platform() == 'brickpi':
15+
from .brickpi import *
16+
else:
17+
# Import ev3 by default, so that it is covered by documentation.
18+
from .ev3 import *

tests/api_tests.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,32 +3,32 @@
33

44
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
55

6-
import ev3dev
6+
import ev3dev.ev3 as ev3
77

8-
ev3dev.Device.DEVICE_ROOT_PATH = os.path.join(os.path.dirname(__file__), 'fake_sys_class')
8+
ev3.Device.DEVICE_ROOT_PATH = os.path.join(os.path.dirname(__file__), 'fake_sys_class')
99

1010
class TestAPI(unittest.TestCase):
1111
def test_device(self):
12-
d = ev3dev.Device('tacho-motor', 'motor*')
12+
d = ev3.Device('tacho-motor', 'motor*')
1313
self.assertTrue(d.connected)
1414

15-
d = ev3dev.Device('tacho-motor', 'motor0')
15+
d = ev3.Device('tacho-motor', 'motor0')
1616
self.assertTrue(d.connected)
1717

18-
d = ev3dev.Device('tacho-motor', 'motor*', driver_name='lego-ev3-m-motor')
18+
d = ev3.Device('tacho-motor', 'motor*', driver_name='lego-ev3-m-motor')
1919
self.assertTrue(d.connected)
2020

21-
d = ev3dev.Device('tacho-motor', 'motor*', port_name='outA')
21+
d = ev3.Device('tacho-motor', 'motor*', port_name='outA')
2222
self.assertTrue(d.connected)
2323

24-
d = ev3dev.Device('tacho-motor', 'motor*', port_name='outA', driver_name='not-valid')
24+
d = ev3.Device('tacho-motor', 'motor*', port_name='outA', driver_name='not-valid')
2525
self.assertTrue(not d.connected)
2626

27-
d = ev3dev.Device('lego-sensor', 'sensor*')
27+
d = ev3.Device('lego-sensor', 'sensor*')
2828
self.assertTrue(d.connected)
2929

3030
def test_medium_motor(self):
31-
m = ev3dev.MediumMotor()
31+
m = ev3.MediumMotor()
3232

3333
self.assertTrue(m.connected);
3434

@@ -52,15 +52,15 @@ def test_medium_motor(self):
5252
self.assertEqual(m.speed, 0)
5353
self.assertEqual(m.speed_regulation_enabled, 'off')
5454
self.assertEqual(m.speed_sp, 0)
55-
self.assertEqual(m.state, [])
55+
self.assertEqual(m.state, ['running'])
5656
self.assertEqual(m.stop_command, 'coast')
5757
self.assertEqual(m.time_sp, 1000)
5858

5959
with self.assertRaises(Exception):
6060
c = m.command
6161

6262
def test_infrared_sensor(self):
63-
s = ev3dev.InfraredSensor()
63+
s = ev3.InfraredSensor()
6464

6565
self.assertTrue(s.connected)
6666

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
1+
running

0 commit comments

Comments
 (0)