Spaces:
Runtime error
Runtime error
File size: 1,602 Bytes
f750b58 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import pytest
import re
from fewshot import FewShot4UAVs
@pytest.fixture
def fewshot():
return FewShot4UAVs()
# format: <command> \t<goal>
def parse_command(text):
tokens = re.split(' \t|\n|: ', text.lower())
print(tokens)
if len(tokens) >= 2:
return tokens[1]
else:
return 'none'
def test_picture(fewshot):
text = "Take a picture."
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
def test_snap_pic(fewshot):
text = "Snap a photo."
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
def test_no_location1(fewshot):
text = "Check out."
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
def test_no_location2(fewshot):
text = "Go to."
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
def test_no_location3(fewshot):
text = "Travel to."
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
def test_no_location4(fewshot):
text = "Head to."
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
def test_empty_input(fewshot):
text = ""
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
def test_invalid_command(fewshot):
text = "Do something."
output = fewshot.get_command(text)
command = parse_command(output)
assert command == "none"
|